(tvm) branch main updated: Fixed neg operator conversion

[tqchen]

This is an automated email from the ASF dual-hosted git repository.

tqchen 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 b09c9195b4 Fixed neg operator conversion
b09c9195b4 is described below

commit b09c9195b4733129bf52a9d2db4ebcdec168b86f
Author: Patrik Persson <patrik.pers...@inceptron.io>
AuthorDate: Fri Dec 20 14:44:12 2024 +0100

    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 f5083caf82..573cde982b 100644
--- a/python/tvm/relax/frontend/onnx/onnx_frontend.py
+++ b/python/tvm/relax/frontend/onnx/onnx_frontend.py
@@ -1526,6 +1526,8 @@ class Neg(OnnxOpConverter):
         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 @@ class Range(OnnxOpConverter):
             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):