tilelang.jit.adapter.base

The profiler and convert to torch utils

Classes

BaseKernelAdapter

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

Module Contents

class tilelang.jit.adapter.base.BaseKernelAdapter(mod, params, result_idx)

Bases: abc.ABC

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

Parameters:

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

• result_idx (List[int])

func: Callable | None = None

mod

params

result_idx

__call__(*args, **kwds)

Parameters:

• args (Any)

• kwds (Any)

Return type:

Any

get_kernel_source()

Return type:

str