Skip to content

GitLab

"examples/nas/git@developer.sourcefind.cn:OpenDAS/nni.git" did not exist on "73b2221b5eb4fd21802e6bf41e21d5df8ef9bf2c"
Commit 68989d80 authored by Lei Wang's avatar Lei Wang Committed by LeiWang1999
Browse files

support torch.bool as kernel input (#636)

parent e937faa6
Hide 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:
torch.int16: ctypes.c_int16,
torch.int32: ctypes.c_int32,
torch.int64: ctypes.c_int64,
torch.bool: ctypes.c_bool,
}
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