未验证 提交 925e84bf 编写于 作者: engineer1109's avatar engineer1109 提交者: GitHub

[chery-pick] Fix namespace error (#45925) (#46029)

修复cuda11.7编译出错的问题
上级 20d168d9
...@@ -71,14 +71,14 @@ __global__ void GroupNormBackwardGetMeanAndVar(const T* x, ...@@ -71,14 +71,14 @@ __global__ void GroupNormBackwardGetMeanAndVar(const T* x,
if (flags & kHasScale) { if (flags & kHasScale) {
#if CUDA_VERSION >= 11070 #if CUDA_VERSION >= 11070
platform::CudaAtomicAdd(&(d_scale[ccid]), d_scale_data); paddle::platform::CudaAtomicAdd(&(d_scale[ccid]), d_scale_data);
#else #else
CudaAtomicAddWithWarp(&(d_scale[ccid]), d_scale_data); CudaAtomicAddWithWarp(&(d_scale[ccid]), d_scale_data);
#endif #endif
} }
if (flags & kHasBias) { if (flags & kHasBias) {
#if CUDA_VERSION >= 11070 #if CUDA_VERSION >= 11070
platform::CudaAtomicAdd(&(d_bias[ccid]), d_bias_data); paddle::platform::CudaAtomicAdd(&(d_bias[ccid]), d_bias_data);
#else #else
CudaAtomicAddWithWarp(&(d_bias[ccid]), d_bias_data); CudaAtomicAddWithWarp(&(d_bias[ccid]), d_bias_data);
#endif #endif
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册