From 2feefbdaf82fe1a98e506ead07b2b671bb412d23 Mon Sep 17 00:00:00 2001 From: yanglf1121 Date: Fri, 9 Jul 2021 16:29:02 +0800 Subject: [PATCH] fix naming collision --- .../backend/kernel_compiler/gpu/cuda_impl/cumprod_impl.cu | 8 ++++---- .../backend/kernel_compiler/gpu/cuda_impl/cumsum_impl.cu | 8 ++++---- 2 files changed, 8 insertions(+), 8 deletions(-) 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 fcf7209c46..2836c781f6 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 1b6ce2f3ec..63019fc927 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); }