tilelang.jit.adapter.cython.cython_wrapper module#
- class tilelang.jit.adapter.cython.cython_wrapper.CythonKernelWrapper#
Bases:
object- forward(inputs, stream=-1, skip_tensor_validation=False)#
- set_buffer_device_map(buffer_device_map)#
- set_buffer_dtype_map(buffer_dtype_map)#
- set_dynamic_symbolic_map(dynamic_symbolic_map)#
- set_ptr_map(ptr_map)#
- set_static_shape_map(static_shape_map)#