From 1c4f13c8f09f5640888e1617c50b01e8d0bb5cdb Mon Sep 17 00:00:00 2001 From: jinjieliu Date: Sun, 8 Feb 2026 16:15:56 +0800 Subject: [PATCH] find packages by sysconfig instead of importlib Signed-off-by: jinjieliu --- python/triton_tvm_ffi/utils.py | 7 +++---- 1 file changed, 3 insertions(+), 4 deletions(-) diff --git a/python/triton_tvm_ffi/utils.py b/python/triton_tvm_ffi/utils.py index 694b438..87d7e34 100644 --- a/python/triton_tvm_ffi/utils.py +++ b/python/triton_tvm_ffi/utils.py @@ -1,13 +1,12 @@ -import importlib.resources -from importlib.resources.abc import Traversable +import sysconfig from typing import List, Optional from triton.backends.nvidia.driver import ty_to_cpp def include_paths() -> List[str]: - pkg_path: Traversable = importlib.resources.files("triton_tvm_ffi") - return [str(pkg_path / "include"), str(pkg_path / "../../include")] + pkg_path: str = sysconfig.get_path("purelib") + return [f"{pkg_path}/triton_tvm_ffi/include"] def type_canonicalize(ty: str) -> Optional[str]: