tilelang.backend.pass_pipeline.pipeline

Attributes

LowerFunc

Classes

PassPipeline

Lowering pass pipeline for a specific backend.

Functions

register_pipeline(pipeline)	Register a lowering pipeline for a backend.
get_pipeline(name)	Return the registered Pipeline for name.
resolve_pipeline(target)	Resolve the lowering pipeline from a TVM target.

Module Contents

tilelang.backend.pass_pipeline.pipeline.LowerFunc

class tilelang.backend.pass_pipeline.pipeline.PassPipeline(name, lower)

Lowering pass pipeline for a specific backend.

Each backend should register its own Pipeline so that the compiler can resolve the correct pass sequence from the target at runtime.

Parameters:

• name (str)

• lower (LowerFunc)

name

lower(mod, target)

Parameters:

• mod (tvm.IRModule)

• target (tvm.target.Target)

Return type:

tvm.IRModule

tilelang.backend.pass_pipeline.pipeline.register_pipeline(pipeline)

Register a lowering pipeline for a backend.

The pipeline name should match target.kind.name (e.g. "cuda", "hip", "c", "llvm").

Parameters:

pipeline (PassPipeline)

Return type:

PassPipeline

tilelang.backend.pass_pipeline.pipeline.get_pipeline(name)

Return the registered Pipeline for name.

Parameters:

name (str)

Return type:

PassPipeline

tilelang.backend.pass_pipeline.pipeline.resolve_pipeline(target)

Resolve the lowering pipeline from a TVM target.

Parameters:

target (tvm.target.Target)

Return type:

PassPipeline