[Fix][ONNX] Fix CumSum conversion when loading ONNX model (apache#18137)

* Fix onnx cumsum

* Fix onnx cumsum

Author: ConvolutedDog

python/tvm/relax/frontend/onnx/onnx_frontend.py

@@ -1338,9 +1338,16 @@ def _impl_v14(cls, bb, inputs, attr, params):
             axis = int(axis.data.numpy())
         elif isinstance(axis, relax.Var):
             axis = 0
+
+        if attr.get("reverse", 0) != 0:
+            data = bb.emit_te(topi.flip, data, axis=axis if axis else 0)
+
         data = relax.op.cumsum(data, axis)
+        data = bb.normalize(data)
+
         if attr.get("reverse", 0) != 0:
             data = bb.emit_te(topi.flip, data, axis=axis if axis else 0)
+
         return data
 
 

tests/python/relax/test_frontend_onnx.py

@@ -1099,7 +1099,7 @@ def test_pow():
     verify_binary("Pow", [32, 32], [32, 32], [32, 32])
 
 
-@pytest.mark.parametrize("reverse", [False])
+@pytest.mark.parametrize("reverse", [True, False])
 @pytest.mark.parametrize("exclusive", [False])
 def test_cumsum(reverse, exclusive):
     cumsum_node = helper.make_node(