This website works better with JavaScript.
Home
Issues
Pull Requests
Milestones
AI流水线
Repositories
Datasets
Forum
实训
竞赛
大数据
AI开发
Register
Sign In
Huawei_Technology
/
mindspore
Not watched
Unwatch
Watch all
Watch but not notify
1
Star
0
Fork
0
Code
Releases
13
Wiki
evaluate
Activity
Issues
0
Pull Requests
0
Datasets
Model
Cloudbrain
HPC
Browse Source
fix codex warning
tags/v0.3.0-alpha
wilfChen
5 years ago
parent
afe048474d
commit
a266fd03b0
2 changed files
with
2 additions
and
2 deletions
Unified View
Diff Options
Show Stats
Download Patch File
Download Diff File
+1
-1
mindspore/ccsrc/kernel/gpu/nn/layer_norm_gpu_kernel.h
+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 mean = GetDeviceAddress<T>(outputs, 1);
auto variance = GetDeviceAddress<T>(outputs, 2);
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,
LayerNorm(input_row_, input_col_, param_dim_, epsilon, x, gamma, beta, y, mean, variance,
reinterpret_cast<cudaStream_t>(stream_ptr));
reinterpret_cast<cudaStream_t>(stream_ptr));
return true;
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 dg = GetDeviceAddress<T>(outputs, 1);
auto db = GetDeviceAddress<T>(outputs, 2);
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,
LayerNormGrad(input_row_, input_col_, param_dim_, epsilon, dy, x, mean, var, gamma, dx, dg, db,
reinterpret_cast<cudaStream_t>(stream_ptr));
reinterpret_cast<cudaStream_t>(stream_ptr));
return true;
return true;
Write
Preview
Loading…
Cancel
Save