提交 a266fd03 编写于 作者: W wilfChen

fix codex warning

上级 afe04847
......@@ -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.
先完成此消息的编辑!
想要评论请 注册