mirror of
https://github.com/sgjzfzzf/triton-tvm-ffi.git
synced 2026-05-02 03:52:11 +08:00
include header files by c/cpp instead of jinja
Signed-off-by: jinjieliu <jinjie.liu@usc.edu>
This commit is contained in:
@@ -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
|
||||
|
||||
Reference in New Issue
Block a user