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

Merge pull request !2696 from zyli2020/bug_fix
5 years ago · fafdfa9a03
parent 6fb5538117 7bbe183e6f
commit fafdfa9a03
1 changed files with 2 additions and 1 deletions
--- a/mindspore/ccsrc/kernel/gpu/arrays/array_reduce_gpu_kernel.h
+++ b/mindspore/ccsrc/kernel/gpu/arrays/array_reduce_gpu_kernel.h
@ -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);