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.

参数:

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

• result_idx (list[int])

• func_or_mod (tvm.tir.PrimFunc | tilelang.tvm.IRModule)

• device_mod (tilelang.tvm.IRModule | None)

• kernel_global_source (str | None)

• verbose (bool)

kernel_global_source = None

kernel_name

verbose = False

block_info = [1, 1, 1]

grid_info = [1, 1, 1]