tilelang.language.cluster¶
Functions¶
Issue barrier.cluster.arrive.relaxed.aligned. |
|
Issue barrier.cluster.arrive.aligned. |
|
Issue barrier.cluster.wait.aligned. |
|
Issue cluster barrier arrive + wait (full synchronization). |
|
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