From cbafd3ed1f618d1cd76554a372a95e64b8e2c093 Mon Sep 17 00:00:00 2001 From: Patrik Persson Date: Fri, 20 Dec 2024 14:44:12 +0100 Subject: [PATCH] Fixed neg operator conversion Updated arange converter to use relax.op.arange instead of emitting a tir arange --- python/tvm/relax/frontend/onnx/onnx_frontend.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/python/tvm/relax/frontend/onnx/onnx_frontend.py b/python/tvm/relax/frontend/onnx/onnx_frontend.py index f5083caf8205..573cde982bea 100644 --- a/python/tvm/relax/frontend/onnx/onnx_frontend.py +++ b/python/tvm/relax/frontend/onnx/onnx_frontend.py @@ -1526,6 +1526,8 @@ def _impl_v13(cls, bb, inputs, attr, params): if isinstance(inputs[0], relax.Constant): data_np = inputs[0].data.numpy() return relax.const(_np.negative(data_np), inputs[0].struct_info.dtype) + if isinstance(inputs[0], relax.PrimValue): + return relax.PrimValue(-inputs[0].value) return relax.op.negative(inputs[0]) @@ -2189,7 +2191,7 @@ def _impl_v12(cls, bb, inputs, attr, params): return relax.const(out_range, out_dtype) # Otherwise compute in graph. - return bb.emit_te(topi.arange, start, limit, step, out_dtype) + return relax.op.arange(start, limit, step, out_dtype) class InstanceNormalization(OnnxOpConverter):