tilelang.jit.adapter.dlpack

The profiler and convert to torch utils

Classes

TorchDLPackKernelAdapter

Helper class that provides a standard way to create an ABC using

Module Contents

class tilelang.jit.adapter.dlpack.TorchDLPackKernelAdapter(mod, params, result_idx)

Bases: tilelang.jit.adapter.base.BaseKernelAdapter

Helper class that provides a standard way to create an ABC using inheritance.

Parameters:

• params (List[tilelang.engine.param.KernelParam])

• result_idx (List[int])