tilelang.rocm.pipeline

Attributes

rocm_pipeline

Functions

ROCMPassPipelineBody(mod, target)

Module Contents

tilelang.rocm.pipeline.ROCMPassPipelineBody(mod, target)

Parameters:

• mod (tvm.IRModule)

• target (tvm.target.Target)

Return type:

tvm.IRModule

tilelang.rocm.pipeline.rocm_pipeline