Skip to content

Commit 0211a67

Browse files
authored
[BC breaking] Add StackTensor support to hl.signal & hl.wait (as_ptrs) (#261)
1 parent 13db75f commit 0211a67

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
@@ -120,10 +120,6 @@ def helion_matmul_w_progress(
120120
tile_m.begin // (M_per_rank // SPLITS_PER_RANK),
121121
],
122122
signal=1,
123-
update=None,
124-
op="ld",
125-
scope="gpu",
126-
sem="acquire",
127123
)
128124
for tile_k in hl.tile(K):
129125
acc = torch.addmm(acc, a[tile_m, tile_k], b[tile_k, tile_n])

0 commit comments

Comments
 (0)