use templates to substitute parts of macros

Signed-off-by: jinjieliu <jinjie.liu@usc.edu>
2026-05-02 03:52:11 +08:00 · 2026-02-08 22:24:12 +08:00
parent 1c4f13c8f0
commit 213e4fc060
9 changed files with 139 additions and 49 deletions
--- a/python/triton_tvm_ffi/jit.py
+++ b/python/triton_tvm_ffi/jit.py
@@ -40,7 +40,8 @@ class TVMFFIJITFunction(object):
            grid: Union[
                Callable[[Dict[str, Any]], Tuple[int, int, int]], Tuple[int, int, int]
            ],
-            _: int,
+            _device: int,
+            _stream: int,
            args: Sequence[Any],
            kwargs: Mapping[str, Any],
        ):