tilelang.language.cluster¶

Functions¶

cluster_arrive_relaxed()

Issue barrier.cluster.arrive.relaxed.aligned.

cluster_arrive()

Issue barrier.cluster.arrive.aligned.

cluster_wait()

Issue barrier.cluster.wait.aligned.

cluster_sync()

Issue cluster barrier arrive + wait (full synchronization).

block_rank_in_cluster()

Return the 1-D rank of the calling CTA within its cluster (%%cluster_ctarank).

Module Contents¶

tilelang.language.cluster.cluster_arrive_relaxed()¶

Issue barrier.cluster.arrive.relaxed.aligned.

Return type:

tvm.tir.PrimExpr

tilelang.language.cluster.cluster_arrive()¶

Issue barrier.cluster.arrive.aligned.

Return type:

tvm.tir.PrimExpr

tilelang.language.cluster.cluster_wait()¶

Issue barrier.cluster.wait.aligned.

Return type:

tvm.tir.PrimExpr

tilelang.language.cluster.cluster_sync()¶

Issue cluster barrier arrive + wait (full synchronization).

Return type:

tvm.tir.PrimExpr

tilelang.language.cluster.block_rank_in_cluster()¶

Return the 1-D rank of the calling CTA within its cluster (%%cluster_ctarank).

Return type:

tvm.tir.PrimExpr