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

@@ -8,6 +8,7 @@ import torch.utils.cpp_extension
import tvm_ffi
from .jit import TVMFFIJITFunction
from .utils import include_paths
class TVMFFIWrapperFunction(object):
@@ -103,7 +104,7 @@ def wrap(
extra_cflags,
extra_cuda_cflags,
extra_ldflags,
extra_include_paths,
include_paths() + (extra_include_paths or []),
)
return decorate