support torch.bool as kernel input (#636)

68989d80 · Lei Wang · LeiWang1999 · e937faa6 · 68989d80
Commit 68989d80 authored Jul 15, 2025 by Lei Wang Committed by LeiWang1999 Jul 15, 2025
Show whitespace changes
Inline Side-by-side

Showing with 1 addition and 0 deletions

tilelang/jit/adapter/cython/cython_wrapper.pyx tilelang/jit/adapter/cython/cython_wrapper.pyx +1 -0

No files found.
--- a/tilelang/jit/adapter/cython/cython_wrapper.pyx
+++ b/tilelang/jit/adapter/cython/cython_wrapper.pyx
@@ -166,6 +166,7 @@ cdef class CythonKernelWrapper:
            torch.int16: ctypes.c_int16,
            torch.int32: ctypes.c_int32,
            torch.int64: ctypes.c_int64,
+            torch.bool: ctypes.c_bool,
        }
        
        call_args = []