!12660 fix gpu bceloss bug

From: @huaweib Reviewed-by: @zhoufeng54,@chujinjin Signed-off-by: @chujinjin
2021-02-27 11:02:36 +08:00 · 2021-02-27 11:02:36 +08:00 · a7bfc16aca
parent 5589881ffe 4a3fcc086b
commit a7bfc16aca
1 changed files with 1 additions and 1 deletions
--- 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();