tilelang.backend.pass_pipeline.pipeline_utils

Functions

allow_vectorize([pass_ctx])
allow_global_thread_synchronization([pass_ctx])
should_enable_aggressive_merge([pass_ctx, target])
should_force_let_inline([pass_ctx])
should_enable_layout_visual([pass_ctx])
should_enable_race_check([pass_ctx])
should_disable_shared_memory_reuse([pass_ctx])
get_layout_visual_formats([pass_ctx])
LayoutVisual(mod)	Apply layout visualization pass if enabled.

Module Contents

tilelang.backend.pass_pipeline.pipeline_utils.allow_vectorize(pass_ctx=None)

Parameters:

pass_ctx (tilelang.transform.PassContext | None)

Return type:

bool

tilelang.backend.pass_pipeline.pipeline_utils.allow_global_thread_synchronization(pass_ctx=None)

Parameters:

pass_ctx (tilelang.transform.PassContext | None)

Return type:

bool

tilelang.backend.pass_pipeline.pipeline_utils.should_enable_aggressive_merge(pass_ctx=None, target=None)

Parameters:

• pass_ctx (tilelang.transform.PassContext | None)

• target (tvm.target.Target | None)

Return type:

bool

tilelang.backend.pass_pipeline.pipeline_utils.should_force_let_inline(pass_ctx=None)

Parameters:

pass_ctx (tilelang.transform.PassContext | None)

Return type:

bool

tilelang.backend.pass_pipeline.pipeline_utils.should_enable_layout_visual(pass_ctx=None)

Parameters:

pass_ctx (tilelang.transform.PassContext | None)

Return type:

bool

tilelang.backend.pass_pipeline.pipeline_utils.should_enable_race_check(pass_ctx=None)

Parameters:

pass_ctx (tilelang.transform.PassContext | None)

Return type:

bool

tilelang.backend.pass_pipeline.pipeline_utils.should_disable_shared_memory_reuse(pass_ctx=None)

Parameters:

pass_ctx (tilelang.transform.PassContext | None)

Return type:

bool

tilelang.backend.pass_pipeline.pipeline_utils.get_layout_visual_formats(pass_ctx=None)

Parameters:

pass_ctx (tilelang.transform.PassContext | None)

Return type:

list[str]

tilelang.backend.pass_pipeline.pipeline_utils.LayoutVisual(mod)

Apply layout visualization pass if enabled.

Parameters:

mod (tvm.IRModule)

Return type:

None