This is an automated email from the ASF dual-hosted git repository.
echuraev pushed a commit to branch main
in repository https://gitbox.apache.org/repos/asf/tvm.git
The following commit(s) were added to refs/heads/main by this push:
new bd906f082f [ONNX] Fix missing helper in AffineGrid test (#19920)
bd906f082f is described below
commit bd906f082f1840afa0de7ebb38146ecfb849afff
Author: Shushi Hong <[email protected]>
AuthorDate: Wed Jul 1 16:58:59 2026 -0400
[ONNX] Fix missing helper in AffineGrid test (#19920)
This pr adds a onnx helper that cleaned by #19880 and fixes ci error
---
tests/python/relax/test_frontend_onnx.py | 11 +++++++++++
1 file changed, 11 insertions(+)
diff --git a/tests/python/relax/test_frontend_onnx.py
b/tests/python/relax/test_frontend_onnx.py
index b05b8e3742..b48cde222d 100644
--- a/tests/python/relax/test_frontend_onnx.py
+++ b/tests/python/relax/test_frontend_onnx.py
@@ -50,6 +50,17 @@ bg = np.random.MT19937(0)
rg = np.random.Generator(bg)
+def collect_relax_call_ops(func):
+ call_ops = set()
+
+ def _visit(expr):
+ if isinstance(expr, relax.Call) and isinstance(expr.op, tvm.ir.Op):
+ call_ops.add(expr.op.name)
+
+ relax.analysis.post_order_visit(func.body, _visit)
+ return call_ops
+
+
def generate_random_inputs(
model: ModelProto, inputs: dict[str, np.ndarray] | None = None
) -> dict[str, np.ndarray]:
