Files
apache--tvm/python
Neo Chien 6569cf0ac9 [Relax][ONNX] Fix CumSum axis handling: support runtime axis tensor, error on multi-element axis (#19467)
Hi Committers,

This PR is trying to fix issues
https://github.com/apache/tvm/issues/19437. Any suggestions would be
appreciated if you are available.

### Root Cause
The original CumSum converter always defaulted to axis=0 when the axis
input was a relax.Var (i.e., a runtime tensor), ignoring the actual
runtime value. This led to incorrect behavior and did not comply with
the ONNX specification.

### Solutions
Update CumSum._impl_v14 to:
- Check if the axis input is a Constant: require it to have exactly one
element, otherwise raise an error.
- If the axis input is a relax.Var, raise an error instead of always
defaulted to axis=0.

---------

Co-authored-by: cchung100m <cchung100m@users.noreply.github.com>
2026-05-01 19:01:42 +08:00
..