fix(fronted): 修复reduce算子对reducesum的opset特判bug

InfiniTensor · Jan 31, 2024 · 3f5b0ee · 3f5b0ee
1 parent eea2249
commit 3f5b0ee
Showing 1 changed file with 14 additions and 4 deletions.
diff --git a/src/07onnx/src/operators/reduce.cc b/src/07onnx/src/operators/reduce.cc
@@ -20,13 +20,23 @@ namespace refactor::onnx {
 
         auto noopWithEmptyAxes = false;
         decltype(Op::axes) axes = std::nullopt;
-        if (opsetVer >= 18) {
-            noopWithEmptyAxes = attributes.getOrInsert( "noop_with_empty_axes", {0}).int_() != 0;
+
+        // 针对ReduceSum做特判
+        if (opType == "onnx::ReduceSum") {
+            if (opsetVer >= 13) {
+                noopWithEmptyAxes = attributes.getOrInsert("noop_with_empty_axes", {0}).int_() != 0;
+            } else {
+                axes.emplace(attributes.getOrInsert("axes", {{}}).ints());
+            }
         } else {
-            axes.emplace(attributes.getOrInsert( "axes", {{}}).ints());
+            if (opsetVer >= 18) {
+                noopWithEmptyAxes = attributes.getOrInsert("noop_with_empty_axes", {0}).int_() != 0;
+            } else {
+                axes.emplace(attributes.getOrInsert("axes", {{}}).ints());
+            }
         }
 
-        auto keepDims = attributes.getOrInsert( "keepdims", {1}).int_();
+        auto keepDims = attributes.getOrInsert("keepdims", {1}).int_();
         Ty ty;
         if (opType == "onnx::ReduceMean") {
             ty = Ty::Mean;