From 1d94d2a9c88186a046c3bc4134dd9d36f84bf2ff Mon Sep 17 00:00:00 2001 From: thais Date: Fri, 19 Sep 2025 11:49:53 -0300 Subject: [PATCH 1/2] Fixing binding for bert --- .../tvm/relax/frontend/torch/exported_program_translator.py | 6 +++++- 1 file changed, 5 insertions(+), 1 deletion(-) diff --git a/python/tvm/relax/frontend/torch/exported_program_translator.py b/python/tvm/relax/frontend/torch/exported_program_translator.py index b489f3e79496..80bd7f4a0b37 100644 --- a/python/tvm/relax/frontend/torch/exported_program_translator.py +++ b/python/tvm/relax/frontend/torch/exported_program_translator.py @@ -715,7 +715,11 @@ def from_exported_program( if tensor_name == spec.target: bind_name = spec.arg.name break - binding[bind_name] = tvm.runtime.from_dlpack(tensor_value.detach()) + try: + binding[bind_name] = tvm.nd.from_dlpack(tensor_value.detach()) + except RuntimeError: + tensor_cpu = tensor_value.detach().cpu().contiguous() + binding[bind_name] = tvm.nd.array(tensor_cpu.numpy()) mod = self.block_builder.get() mod = relax.transform.BindParams("main", binding)(mod) From fae53907a0bf905b55bdd36cfa47073a914a6383 Mon Sep 17 00:00:00 2001 From: thais Date: Fri, 19 Sep 2025 15:06:12 -0300 Subject: [PATCH 2/2] Fixing names --- .../tvm/relax/frontend/torch/exported_program_translator.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/python/tvm/relax/frontend/torch/exported_program_translator.py b/python/tvm/relax/frontend/torch/exported_program_translator.py index 80bd7f4a0b37..7c20d1b1a469 100644 --- a/python/tvm/relax/frontend/torch/exported_program_translator.py +++ b/python/tvm/relax/frontend/torch/exported_program_translator.py @@ -716,10 +716,10 @@ def from_exported_program( bind_name = spec.arg.name break try: - binding[bind_name] = tvm.nd.from_dlpack(tensor_value.detach()) + binding[bind_name] = tvm.runtime.from_dlpack(tensor_value.detach()) except RuntimeError: tensor_cpu = tensor_value.detach().cpu().contiguous() - binding[bind_name] = tvm.nd.array(tensor_cpu.numpy()) + binding[bind_name] = tvm.runtime.tensor(tensor_cpu.numpy()) mod = self.block_builder.get() mod = relax.transform.BindParams("main", binding)(mod)