Browse Source

optimize error about BroadcastOpGrad

tags/v1.6.0
tacyi139 4 years ago
parent
commit
7413775490
1 changed files with 2 additions and 1 deletions
  1. +2
    -1
      mindspore/ccsrc/backend/kernel_compiler/gpu/math/broadcast_grad_gpu_kernel.h

+ 2
- 1
mindspore/ccsrc/backend/kernel_compiler/gpu/math/broadcast_grad_gpu_kernel.h View File

@@ -98,7 +98,8 @@ class BroadcastOpGradGpuKernel : public GpuKernel {
x1_shape_[i + x1_offset] = shape1[i];
} else {
auto index = i + x1_offset;
MS_LOG(EXCEPTION) << "Invalid input1 index: " << index;
MS_LOG(EXCEPTION) << "For 'BroadcastOpGrad', the dimension of input cannot be greater than " << kMaxShapeSize
<< ", but got " << (index + 1);
}
}
input1_num_ *= shape1[i];


Loading…
Cancel
Save