提交 68f80d5d 编写于 作者: M mindspore-ci-bot 提交者: Gitee

!942 fix codex warning

Merge pull request !942 from chenweifeng/codex
......@@ -43,7 +43,7 @@ class LayerNormGpuKernel : public GpuKernel {
auto mean = GetDeviceAddress<T>(outputs, 1);
auto variance = GetDeviceAddress<T>(outputs, 2);
T epsilon = 10e-12;
const T epsilon = 10e-12;
LayerNorm(input_row_, input_col_, param_dim_, epsilon, x, gamma, beta, y, mean, variance,
reinterpret_cast<cudaStream_t>(stream_ptr));
return true;
......
......@@ -45,7 +45,7 @@ class LayerNormGradGpuKernel : public GpuKernel {
auto dg = GetDeviceAddress<T>(outputs, 1);
auto db = GetDeviceAddress<T>(outputs, 2);
T epsilon = 10e-12;
const T epsilon = 10e-12;
LayerNormGrad(input_row_, input_col_, param_dim_, epsilon, dy, x, mean, var, gamma, dx, dg, db,
reinterpret_cast<cudaStream_t>(stream_ptr));
return true;
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册