tilelang.contrib.cutedsl.cluster¶

Functions¶

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_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