Browse Source

fix codex warning

tags/v0.3.0-alpha
wilfChen 5 years ago
parent
commit
a266fd03b0
2 changed files with 2 additions and 2 deletions
  1. +1
    -1
      mindspore/ccsrc/kernel/gpu/nn/layer_norm_gpu_kernel.h
  2. +1
    -1
      mindspore/ccsrc/kernel/gpu/nn/layer_norm_grad_gpu_kernel.h

+ 1
- 1
mindspore/ccsrc/kernel/gpu/nn/layer_norm_gpu_kernel.h View File

@@ -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;


+ 1
- 1
mindspore/ccsrc/kernel/gpu/nn/layer_norm_grad_gpu_kernel.h View File

@@ -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;


Loading…
Cancel
Save