Nigel
/
mindspore
forked from mindspore-Ecosystem/mindspore
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

!5490 modify the format info of tensorAdd 

Merge pull request !5490 from limingqi107/r0.7
This commit is contained in:
mindspore-ci-bot 2020-09-03 11:07:12 +08:00 committed by Gitee
parent a585177825 0775db0940
commit 40bf2493fa
1 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
mindspore/ccsrc/backend/kernel_compiler/gpu/math/broadcast_gpu_kernel.h
View File

@ -56,9 +56,9 @@ class BroadcastOpGpuKernel : public GpuKernel {
} }
bool Init(const CNodePtr &kernel_node) override { bool Init(const CNodePtr &kernel_node) override {
GetOpType(kernel_node); GetOpType(kernel_node);
auto shape1 = AnfAlgo::GetPrevNodeOutputInferShape(kernel_node, 0); auto shape1 = AnfAlgo::GetInputDeviceShape(kernel_node, 0);
auto shape2 = AnfAlgo::GetPrevNodeOutputInferShape(kernel_node, 1); auto shape2 = AnfAlgo::GetInputDeviceShape(kernel_node, 1);
auto shape3 = AnfAlgo::GetOutputInferShape(kernel_node, 0); auto shape3 = AnfAlgo::GetOutputDeviceShape(kernel_node, 0);
need_broadcast_ = IsBroadcast(shape1, shape2); need_broadcast_ = IsBroadcast(shape1, shape2);
if (need_broadcast_ && shape1.size() > 7) { if (need_broadcast_ && shape1.size() > 7) {
MS_LOG(EXCEPTION) << "Broadcast operation not support dim greater than 7"; MS_LOG(EXCEPTION) << "Broadcast operation not support dim greater than 7";