JinjieLiu/triton-tvm-ffi
1
0
Fork 0
mirror of https://github.com/sgjzfzzf/triton-tvm-ffi.git synced 2026-05-02 03:52:11 +08:00
Code Issues Packages Projects Releases Wiki Activity

find packages by sysconfig instead of importlib

Browse Source 
Signed-off-by: jinjieliu <jinjie.liu@usc.edu>
This commit is contained in:
jinjieliu
2026-02-08 16:15:56 +08:00
parent 24237a6313
commit 1c4f13c8f0
1 changed files with 3 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
python/triton_tvm_ffi/utils.py
View File

@@ -1,13 +1,12 @@
import importlib.resources import sysconfig
from importlib.resources.abc import Traversable
from typing import List, Optional from typing import List, Optional
from triton.backends.nvidia.driver import ty_to_cpp from triton.backends.nvidia.driver import ty_to_cpp
def include_paths() -> List[str]: def include_paths() -> List[str]:
pkg_path: Traversable = importlib.resources.files("triton_tvm_ffi") pkg_path: str = sysconfig.get_path("purelib")
return [str(pkg_path / "include"), str(pkg_path / "../../include")] return [f"{pkg_path}/triton_tvm_ffi/include"]
def type_canonicalize(ty: str) -> Optional[str]: def type_canonicalize(ty: str) -> Optional[str]: