From 156afadb28bf80c5af90c1aaef145c4407cfb51b Mon Sep 17 00:00:00 2001 From: buxue Date: Mon, 12 Jul 2021 09:56:49 +0800 Subject: [PATCH] fix type bug in fp16 PReLU grad --- .../backend/kernel_compiler/gpu/cuda_impl/prelu_grad_impl.cu | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/mindspore/ccsrc/backend/kernel_compiler/gpu/cuda_impl/prelu_grad_impl.cu b/mindspore/ccsrc/backend/kernel_compiler/gpu/cuda_impl/prelu_grad_impl.cu index ac5b73d948f..2b2cd7d5dbc 100644 --- a/mindspore/ccsrc/backend/kernel_compiler/gpu/cuda_impl/prelu_grad_impl.cu +++ b/mindspore/ccsrc/backend/kernel_compiler/gpu/cuda_impl/prelu_grad_impl.cu @@ -41,7 +41,7 @@ __global__ void InitDwArrayData(size_t dw_array_size, float *dw_array) { template __global__ void ComputeDwData(size_t weight_size, size_t thread_num, const float *dw_array, T *dw) { for (size_t i = blockIdx.x * blockDim.x + threadIdx.x; i < weight_size; i += blockDim.x * gridDim.x) { - T value = 0.0; + float value = 0.0; for (size_t j = 0; j < thread_num; j++) { value += dw_array[i * thread_num + j]; }