tilelang.metal.pipeline

Attributes

metal_pipeline

Functions

MetalPassPipelineBody(mod, target)

Module Contents

tilelang.metal.pipeline.MetalPassPipelineBody(mod, target)

Parameters:

• mod (tvm.IRModule)

• target (tvm.target.Target)

Return type:

tvm.IRModule

tilelang.metal.pipeline.metal_pipeline