(tvm) branch main updated: [Relax][Frontend][NN] Fix SourceModule include resolution for installed wheels (#19822)

[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 d321035443 [Relax][Frontend][NN] Fix SourceModule include resolution 
for installed wheels (#19822)
d321035443 is described below

commit d32103544392861a6743b837faadf93ba08db9e6
Author: Shushi Hong <820958...@qq.com>
AuthorDate: Wed Jun 17 20:39:00 2026 -0400

    [Relax][Frontend][NN] Fix SourceModule include resolution for installed 
wheels (#19822)
    
    `SourceModule.get_includes()` hardcoded
    `tvm_home()/3rdparty/tvm-ffi/...` paths that only exist in a source
    checkout. In a wheel the tvm-ffi/DLPack headers ship in the separate
    `tvm_ffi` package, so the existence assert failed and `nn.SourceModule`
    could not compile extern sources.
    
    This pr resolves includes via libinfo (`libinfo.find_include_path()`,
    `tvm_ffi.libinfo.find_include_path()`, `find_dlpack_include_path()`),
    matching `runtime/module.py` and `rpc/minrpc.py`.
---
 python/tvm/relax/frontend/nn/extern.py | 13 ++++++++-----
 1 file changed, 8 insertions(+), 5 deletions(-)

diff --git a/python/tvm/relax/frontend/nn/extern.py 
b/python/tvm/relax/frontend/nn/extern.py
index e424554367..866da6c4ab 100644
--- a/python/tvm/relax/frontend/nn/extern.py
+++ b/python/tvm/relax/frontend/nn/extern.py
@@ -24,7 +24,9 @@ import tempfile
 from collections.abc import Callable
 from pathlib import Path
 
-from tvm import tirx
+import tvm_ffi
+
+from tvm import libinfo, tirx
 from tvm.runtime import Module, load_static_library
 from tvm.support import cc as _cc
 
@@ -306,15 +308,16 @@ class SourceModule(ExternModule):  # pylint: 
disable=too-few-public-methods
         includes : List[pathlib.Path]
             The list of include paths.
         """
-        tvm_home = SourceModule.tvm_home()
         results = [
-            tvm_home / "include",
-            tvm_home / "3rdparty/tvm-ffi/include",
-            tvm_home / "3rdparty/tvm-ffi/3rdparty/dlpack/include",
+            Path(libinfo.find_include_path()),
+            Path(tvm_ffi.libinfo.find_include_path()),
+            Path(tvm_ffi.libinfo.find_dlpack_include_path()),
         ]
         if tvm_pkg:
+            tvm_home = SourceModule.tvm_home()
             for relative in tvm_pkg:
                 results.append(tvm_home / "3rdparty" / relative)
+        results = list(dict.fromkeys(results))
         for path in results:
             assert path.exists(), f"Not found: {path!s}"
             assert path.is_dir(), f"Not a directory: {path!s}"