tilelang.jit.adapter.torch.metal

Classes

MetalKernelAdapter

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

Module Contents

class tilelang.jit.adapter.torch.metal.MetalKernelAdapter(params, result_idx, func_or_mod, device_mod=None, kernel_global_source=None, verbose=False)

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])

• func_or_mod (Union[tvm.tir.PrimFunc, tilelang.tvm.IRModule])

• device_mod (Optional[tilelang.tvm.IRModule])

• kernel_global_source (Optional[str])

• verbose (bool)

kernel_global_source = None

kernel_name

verbose = False

block_info = [1, 1, 1]

grid_info = [1, 1, 1]