This is an automated email from the ASF dual-hosted git repository.
tlopex 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 97d78aa9eb [Relax][PyTorch] Add `mul` operator in ExportedProgram
frontend (#18496)
97d78aa9eb is described below
commit 97d78aa9eb162c29916c50f6c0534616f1f6d9fe
Author: Masahiro Hiramori <[email protected]>
AuthorDate: Mon Nov 24 14:09:04 2025 +0900
[Relax][PyTorch] Add `mul` operator in ExportedProgram frontend (#18496)
As per title.
---
python/tvm/relax/frontend/torch/exported_program_translator.py | 1 +
1 file changed, 1 insertion(+)
diff --git a/python/tvm/relax/frontend/torch/exported_program_translator.py
b/python/tvm/relax/frontend/torch/exported_program_translator.py
index 883be88837..ac79024acf 100644
--- a/python/tvm/relax/frontend/torch/exported_program_translator.py
+++ b/python/tvm/relax/frontend/torch/exported_program_translator.py
@@ -1151,6 +1151,7 @@ class ExportedProgramImporter(BaseFXGraphImporter):
"minimum.default": self._binary_op(relax.op.minimum,
torch.minimum),
"remainder.Tensor": self._binary_op(relax.op.floor_mod,
operator.mod),
"remainder.Scalar": self._binary_op(relax.op.floor_mod,
operator.mod),
+ "mul": self._binary_op(relax.op.multiply, operator.mul),
"mul.Tensor": self._binary_op(relax.op.multiply, operator.mul),
"mul.Scalar": self._binary_op(relax.op.multiply, operator.mul),
"mul_.Tensor": self._binary_op(relax.op.multiply, operator.mul),
