tilelang.jit.adapter.dlpack module#
The profiler and convert to torch utils
- class tilelang.jit.adapter.dlpack.TorchDLPackKernelAdapter(mod, params: List[KernelParam], result_idx: List[int])#
Bases:
BaseKernelAdapter
The profiler and convert to torch utils
Bases: BaseKernelAdapter