tilelang.contrib.cutedsl.cluster

Functions

cluster_store(ptr, value, dst_cta, index, *[, loc, ip])	Store one value into a peer CTA's shared memory (mapa + store).
cluster_arrive_relaxed(*[, loc, ip])
cluster_arrive(*[, loc, ip])
cluster_wait(*[, loc, ip])
cluster_sync(*[, loc, ip])
block_rank_in_cluster(*[, loc, ip])
clc_try_cancel(result_ptr, mbar_ptr, *[, loc, ip])
clc_try_cancel_multicast(result_ptr, mbar_ptr, *[, ...])
clc_is_canceled(result_ptr, *[, loc, ip])
clc_get_first_ctaid_x(result_ptr, *[, loc, ip])
clc_get_first_ctaid_y(result_ptr, *[, loc, ip])
clc_get_first_ctaid_z(result_ptr, *[, loc, ip])

Module Contents

tilelang.contrib.cutedsl.cluster.cluster_store(ptr, value, dst_cta, index, *, loc=None, ip=None)

Store one value into a peer CTA’s shared memory (mapa + store).

Parameters:

• ptr (cutlass.cute.typing.Pointer)

• dst_cta (cutlass.cute.typing.Int32)

• index (cutlass.cute.typing.Int32)

Return type:

None

tilelang.contrib.cutedsl.cluster.cluster_arrive_relaxed(*, loc=None, ip=None)

Return type:

None

tilelang.contrib.cutedsl.cluster.cluster_arrive(*, loc=None, ip=None)

Return type:

None

tilelang.contrib.cutedsl.cluster.cluster_wait(*, loc=None, ip=None)

Return type:

None

tilelang.contrib.cutedsl.cluster.cluster_sync(*, loc=None, ip=None)

Return type:

None

tilelang.contrib.cutedsl.cluster.block_rank_in_cluster(*, loc=None, ip=None)

Return type:

cutlass.cute.typing.Int32

tilelang.contrib.cutedsl.cluster.clc_try_cancel(result_ptr, mbar_ptr, *, loc=None, ip=None)

Parameters:

• result_ptr (cutlass.cute.typing.Pointer)

• mbar_ptr (cutlass.cute.typing.Pointer)

Return type:

None

tilelang.contrib.cutedsl.cluster.clc_try_cancel_multicast(result_ptr, mbar_ptr, *, loc=None, ip=None)

Parameters:

• result_ptr (cutlass.cute.typing.Pointer)

• mbar_ptr (cutlass.cute.typing.Pointer)

Return type:

None

tilelang.contrib.cutedsl.cluster.clc_is_canceled(result_ptr, *, loc=None, ip=None)

Parameters:

result_ptr (cutlass.cute.typing.Pointer)

Return type:

cutlass.cute.typing.Int32

tilelang.contrib.cutedsl.cluster.clc_get_first_ctaid_x(result_ptr, *, loc=None, ip=None)

Parameters:

result_ptr (cutlass.cute.typing.Pointer)

Return type:

cutlass.cute.typing.Int32

tilelang.contrib.cutedsl.cluster.clc_get_first_ctaid_y(result_ptr, *, loc=None, ip=None)

Parameters:

result_ptr (cutlass.cute.typing.Pointer)

Return type:

cutlass.cute.typing.Int32

tilelang.contrib.cutedsl.cluster.clc_get_first_ctaid_z(result_ptr, *, loc=None, ip=None)

Parameters:

result_ptr (cutlass.cute.typing.Pointer)

Return type:

cutlass.cute.typing.Int32