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). |
|
|
Issue a single-CTA cluster launch control query. |
|
Issue a cluster-wide multicast cluster launch control query. |
|
Return 1 when the CLC query successfully canceled a future launch. |
|
Return the x coordinate of the first CTA in a successful CLC response. |
|
Return the y coordinate of the first CTA in a successful CLC response. |
|
Return the z coordinate of the first CTA in a successful CLC response. |
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
- tilelang.language.cluster.clc_try_cancel(result, mbarrier)¶
Issue a single-CTA cluster launch control query.
- Return type:
tvm.tir.PrimExpr
- tilelang.language.cluster.clc_try_cancel_multicast(result, mbarrier)¶
Issue a cluster-wide multicast cluster launch control query.
- Return type:
tvm.tir.PrimExpr
- tilelang.language.cluster.clc_is_canceled(result)¶
Return 1 when the CLC query successfully canceled a future launch.
- Return type:
tvm.tir.PrimExpr
- tilelang.language.cluster.clc_get_first_ctaid_x(result)¶
Return the x coordinate of the first CTA in a successful CLC response.
- Return type:
tvm.tir.PrimExpr
- tilelang.language.cluster.clc_get_first_ctaid_y(result)¶
Return the y coordinate of the first CTA in a successful CLC response.
- Return type:
tvm.tir.PrimExpr
- tilelang.language.cluster.clc_get_first_ctaid_z(result)¶
Return the z coordinate of the first CTA in a successful CLC response.
- Return type:
tvm.tir.PrimExpr