SIGN IN SIGN UP

[Enhancement] Expose cluster_mask on T.tma_copy (#2932)

* [Enhancement] Expose cluster_mask on T.tma_copy

TMA multicast already composes with tma_copy's split-phase contract: in
LowerBulkCopy, use_multicast is orthogonal to the flag that suppresses the
wait, so annotations={"cluster_mask": ...} produces an issue-only multicast
today. Only the named parameter was missing.

Promote it to a documented keyword argument, matching the parameter that
copy_cluster already exposes under the same name, and write down the
semantics: the lowest-ranked CTA in the mask issues the multicast, other
in-mask CTAs issue nothing, a CTA outside the mask falls back to a unicast
load, and every CTA runs its own expect_tx against its local barrier.

No lowering change.

* [Testing] Drop the duplicated TMA store arrive/wait

T.copy already lowers shared -> global to tma_store + tma_store_arrive +
tma_store_wait, so the explicit pair after it emitted a second, redundant
arrive/wait.
D
Diao Han committed
6af825e072d305a87ac039be0abd27b83c9a1bdc
Parent: 78a4ff6
Committed by GitHub <noreply@github.com> on 8/11/2026, 12:12:38 PM