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).

clc_try_cancel(result, mbarrier)

Issue a single-CTA cluster launch control query.

clc_try_cancel_multicast(result, mbarrier)

Issue a cluster-wide multicast cluster launch control query.

clc_is_canceled(result)

Return 1 when the CLC query successfully canceled a future launch.

clc_get_first_ctaid_x(result)

Return the x coordinate of the first CTA in a successful CLC response.

clc_get_first_ctaid_y(result)

Return the y coordinate of the first CTA in a successful CLC response.

clc_get_first_ctaid_z(result)

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