diff --git a/mindspore/ccsrc/backend/kernel_compiler/gpu/nn/binary_cross_entropy_gpu_kernel.h b/mindspore/ccsrc/backend/kernel_compiler/gpu/nn/binary_cross_entropy_gpu_kernel.h index e3764185d8..3ce4642017 100644 --- a/mindspore/ccsrc/backend/kernel_compiler/gpu/nn/binary_cross_entropy_gpu_kernel.h +++ b/mindspore/ccsrc/backend/kernel_compiler/gpu/nn/binary_cross_entropy_gpu_kernel.h @@ -59,7 +59,7 @@ class BinaryCrossEntropyGpuKernel : public GpuKernel { reduction_ = 2; } workspace_size_ = sizeof(T); - if (reduction_ == 0) { + if (reduction_ != 0) { workspace_size_ *= input_size_; } InitSizeLists();