diff --git a/mindspore/ccsrc/backend/kernel_compiler/gpu/cuda_impl/cumprod_impl.cu b/mindspore/ccsrc/backend/kernel_compiler/gpu/cuda_impl/cumprod_impl.cu index fcf7209c469..2836c781f60 100644 --- a/mindspore/ccsrc/backend/kernel_compiler/gpu/cuda_impl/cumprod_impl.cu +++ b/mindspore/ccsrc/backend/kernel_compiler/gpu/cuda_impl/cumprod_impl.cu @@ -26,7 +26,7 @@ __global__ void Copy(T *input, T *output, size_t size) { } template -__global__ void LeftMove(const T *input, T *output, size_t dim0, size_t dim1, size_t dim2, size_t stride, +__global__ void LeftMoveProd(const T *input, T *output, size_t dim0, size_t dim1, size_t dim2, size_t stride, size_t stride2) { size_t num = dim0 * dim2; size_t i, k, offset; @@ -48,7 +48,7 @@ __global__ void LeftMove(const T *input, T *output, size_t dim0, size_t dim1, si } template -__global__ void RightMove(const T *input, T *output, size_t dim0, size_t dim1, size_t dim2, size_t stride, +__global__ void RightMoveProd(const T *input, T *output, size_t dim0, size_t dim1, size_t dim2, size_t stride, size_t stride2) { size_t num = dim0 * dim2; size_t i, k, offset; @@ -117,12 +117,12 @@ void CumProd(const T *input, T *output, T *workspace, size_t dim0, size_t dim1, int size = dim0 * dim2; if (exclusive_) { if (reverse_) { - RightMove<<>>(input, output, dim0, dim1, dim2, stride, stride2); + RightMoveProd<<>>(input, output, dim0, dim1, dim2, stride, stride2); Copy<<>>(workspace, output, size * dim1); CumProdKernelReverse<<>>(workspace, output, dim0, dim1, dim2, stride, stride2); } else { - LeftMove<<>>(input, output, dim0, dim1, dim2, stride, stride2); + LeftMoveProd<<>>(input, output, dim0, dim1, dim2, stride, stride2); Copy<<>>(workspace, output, size * dim1); CumProdKernel<<>>(workspace, output, dim0, dim1, dim2, stride, stride2); } diff --git a/mindspore/ccsrc/backend/kernel_compiler/gpu/cuda_impl/cumsum_impl.cu b/mindspore/ccsrc/backend/kernel_compiler/gpu/cuda_impl/cumsum_impl.cu index 1b6ce2f3ec6..63019fc9278 100644 --- a/mindspore/ccsrc/backend/kernel_compiler/gpu/cuda_impl/cumsum_impl.cu +++ b/mindspore/ccsrc/backend/kernel_compiler/gpu/cuda_impl/cumsum_impl.cu @@ -26,7 +26,7 @@ __global__ void Copy(T *input, T *output, size_t size) { } template -__global__ void LeftMove(const T *input, T *output, size_t dim0, size_t dim1, size_t dim2, size_t stride, +__global__ void LeftMoveSum(const T *input, T *output, size_t dim0, size_t dim1, size_t dim2, size_t stride, size_t stride2) { size_t num = dim0 * dim2; size_t i, k, offset; @@ -48,7 +48,7 @@ __global__ void LeftMove(const T *input, T *output, size_t dim0, size_t dim1, si } template -__global__ void RightMove(const T *input, T *output, size_t dim0, size_t dim1, size_t dim2, size_t stride, +__global__ void RightMoveSum(const T *input, T *output, size_t dim0, size_t dim1, size_t dim2, size_t stride, size_t stride2) { size_t num = dim0 * dim2; size_t i, k, offset; @@ -117,12 +117,12 @@ void CumSum(const T *input, T *output, T *workspace, size_t dim0, size_t dim1, s int size = dim0 * dim2; if (exclusive_) { if (reverse_) { - RightMove<<>>(input, output, dim0, dim1, dim2, stride, stride2); + RightMoveSum<<>>(input, output, dim0, dim1, dim2, stride, stride2); Copy<<>>(workspace, output, size * dim1); CumSumKernelReverse<<>>(workspace, output, dim0, dim1, dim2, stride, stride2); } else { - LeftMove<<>>(input, output, dim0, dim1, dim2, stride, stride2); + LeftMoveSum<<>>(input, output, dim0, dim1, dim2, stride, stride2); Copy<<>>(workspace, output, size * dim1); CumSumKernel<<>>(workspace, output, dim0, dim1, dim2, stride, stride2); }