Skip to content

GitLab

Commit 68989d80 authored by Lei Wang's avatar Lei Wang Committed by LeiWang1999
Browse files

support torch.bool as kernel input (#636)

parent e937faa6
Show whitespace changes
Inline Side-by-side
Showing with 1 addition and 0 deletions
tilelang/jit/adapter/cython/cython_wrapper.pyx
View file @ 68989d80
...@@ -166,6 +166,7 @@ cdef class CythonKernelWrapper: ...@@ -166,6 +166,7 @@ cdef class CythonKernelWrapper:
torch.int16: ctypes.c_int16, torch.int16: ctypes.c_int16,
torch.int32: ctypes.c_int32, torch.int32: ctypes.c_int32,
torch.int64: ctypes.c_int64, torch.int64: ctypes.c_int64,
torch.bool: ctypes.c_bool,
} }
call_args = [] call_args = []
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment