(tvm) 04/06: Fix the system lib in build tir

[tq...@apache.org]

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

tqchen pushed a commit to branch refactor
in repository https://gitbox.apache.org/repos/asf/tvm.git

commit 906ca1439f733f6e8ed523a65b6cc546bd2eda98
Author: tqchen <ti...@gmail.com>
AuthorDate: Fri Feb 14 19:06:11 2025 -0500

    Fix the system lib in build tir
---
 python/tvm/driver/build_module.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/python/tvm/driver/build_module.py b/python/tvm/driver/build_module.py
index 08af27e32f..fb325de1d3 100644
--- a/python/tvm/driver/build_module.py
+++ b/python/tvm/driver/build_module.py
@@ -257,7 +257,7 @@ def build(
             target_input_mod = {}
             for tgt in target_mod.keys():
                 tir_mod = tvm.IRModule(target_mod[tgt])
-                tir_mod.with_attrs(input_mod.attrs)
+                tir_mod = tir_mod.with_attrs(input_mod.attrs)
                 target_input_mod[tgt] = tir_mod
         else:
             target_input_mod = {target: input_mod}