Skip to content

Commit d7682aa

Browse files
committed
[BC breaking] Add MulticastTensor support to hl.signal & hl.wait (as_ptrs)
stack-info: PR: #261, branch: joydddd/stack/13
1 parent 4e16c94 commit d7682aa

File tree

6 files changed

+326
-147
lines changed

6 files changed

+326
-147
lines changed

examples/all_gather_matmul.py

Lines changed: 0 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -96,10 +96,6 @@ def helion_matmul_w_progress(
9696
tile_m.begin // (M_per_rank // SPLITS_PER_RANK),
9797
],
9898
signal=1,
99-
update=None,
100-
op="ld",
101-
scope="gpu",
102-
sem="acquire",
10399
)
104100
for tile_k in hl.tile(K):
105101
# TODO(joydddd): use a_shared and skip barrier when data is available on local rank.

0 commit comments

Comments
 (0)