(tvm) branch main updated: [BugFix] Fixing binding for bert (#18324)

[tlopex]

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 af82187485 [BugFix] Fixing binding for bert (#18324)
af82187485 is described below

commit af821874857e92743862aeaf59467d176be69999
Author: Thais Camacho <thaiscamac...@gmail.com>
AuthorDate: Fri Sep 19 17:29:14 2025 -0300

    [BugFix] Fixing binding for bert (#18324)
    
    * Fixing binding for bert
    
    * Fixing names
---
 python/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 b489f3e794..7c20d1b1a4 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 @@ class ExportedProgramImporter(BaseFXGraphImporter):
                 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.runtime.from_dlpack(tensor_value.detach())
+            except RuntimeError:
+                tensor_cpu = tensor_value.detach().cpu().contiguous()
+                binding[bind_name] = tvm.runtime.tensor(tensor_cpu.numpy())
 
         mod = self.block_builder.get()
         mod = relax.transform.BindParams("main", binding)(mod)