include header files by c/cpp instead of jinja

Signed-off-by: jinjieliu <jinjie.liu@usc.edu>
This commit is contained in:
jinjieliu
2026-02-07 17:16:49 +08:00
parent 6a19a6b06d
commit 24237a6313
8 changed files with 26 additions and 16 deletions

View File

@@ -1,8 +1,15 @@
from typing import Optional
import importlib.resources
from importlib.resources.abc import Traversable
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")]
def type_canonicalize(ty: str) -> Optional[str]:
if ty == "constexpr":
return None