timekiller
/
mindspore2022
forked from huawei/mindspore2022
0
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

!2696 [bugfix]:Attr of axis of ReduceSum is invalid 

Merge pull request !2696 from zyli2020/bug_fix
This commit is contained in:
mindspore-ci-bot 2020-06-29 15:19:24 +08:00 committed by Gitee
parent 6fb5538117 7bbe183e6f
commit fafdfa9a03
1 changed files with 2 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
mindspore/ccsrc/kernel/gpu/arrays/array_reduce_gpu_kernel.h
View File

@ -94,7 +94,8 @@ class ArrayReduceGpuKernel : public GpuKernel {
}
int input_dim_length = SizeToInt(AnfAlgo::GetPrevNodeOutputInferShape(kernel_node, 0).size());
if (AnfAlgo::GetCNodePrimitive(kernel_node)->GetAttr("axis")->isa<ValueTuple>()) {
if (AnfAlgo::GetCNodePrimitive(kernel_node)->GetAttr("axis")->isa<ValueTuple>() ||
AnfAlgo::GetCNodePrimitive(kernel_node)->GetAttr("axis")->isa<ValueList>()) {
auto attr_axis = GetAttr<std::vector<int>>(kernel_node, "axis");
if (attr_axis.empty()) {
axis_.push_back(-1);