Skip to content

Commit 4a2767a

Browse files
committed
Bump ACLSHMEM to v1.5.0.rc1
Pick up the aclshmem_finalize spelling fix and update Ascend tests/tutorials to use the corrected APIs.
1 parent 4200ee6 commit 4a2767a

8 files changed

Lines changed: 8 additions & 8 deletions

File tree

3rdparty/shmem

Submodule shmem updated from 89f42fa to 81c95ba

python/triton_dist/test/ascend/test_my_pe.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -50,7 +50,7 @@ def run_test_distributed(rank, world_size):
5050
y_ref = torch_mype(x0, rank, 3)
5151
_shmemx_get_my_pe[3, 1, 1](x0)
5252
assert torch.equal(x0.cpu(), y_ref.cpu())
53-
_ = ash.aclshmem_finialize()
53+
_ = ash.aclshmem_finalize()
5454

5555

5656
@pytest.mark.dist

python/triton_dist/test/ascend/test_num_ranks.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -50,7 +50,7 @@ def run_test_distributed(rank, world_size):
5050
y_ref = torch_num_ranks(x0, world_size, 3)
5151
_shmemx_get_num_ranks[3, 1, 1](x0)
5252
assert torch.equal(x0.cpu(), y_ref.cpu())
53-
_ = ash.aclshmem_finialize()
53+
_ = ash.aclshmem_finalize()
5454

5555

5656
@pytest.mark.dist

python/triton_dist/test/ascend/test_symm_at.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -72,7 +72,7 @@ def run_test_distributed(rank, world_size):
7272

7373
# Cleanup
7474
ash.aclshmem_free_tensor(remote_mem)
75-
_ = ash.aclshmem_finialize()
75+
_ = ash.aclshmem_finalize()
7676

7777

7878
@pytest.mark.dist

python/triton_dist/test/ascend/test_wait_notify.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -92,7 +92,7 @@ def run_test_distributed(rank, world_size):
9292
# Cleanup
9393
ash.aclshmem_free_tensor(data_mem)
9494
ash.aclshmem_free_tensor(signal_mem)
95-
_ = ash.aclshmem_finialize()
95+
_ = ash.aclshmem_finalize()
9696

9797

9898
@pytest.mark.dist

tutorials/ascend/01-ascend-allgather-gemm.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -333,7 +333,7 @@ def run_test_distributed():
333333
raise AssertionError(error_msg)
334334

335335
ash.aclshmem_free_tensor(peer_mem)
336-
_ = ash.aclshmem_finialize()
336+
_ = ash.aclshmem_finalize()
337337

338338

339339
if __name__ == "__main__":

tutorials/ascend/02-ascend-gemm-reduce-scatter.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -347,7 +347,7 @@ def run_test_distributed():
347347
raise AssertionError(error_msg)
348348

349349
ash.aclshmem_free_tensor(peer_mem)
350-
_ = ash.aclshmem_finialize()
350+
_ = ash.aclshmem_finalize()
351351

352352

353353
if __name__ == "__main__":

tutorials/ascend/03-ascend-gemm-allreduce-oneshot.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -360,7 +360,7 @@ def run_test_distributed():
360360
raise AssertionError(error_msg)
361361

362362
ash.aclshmem_free_tensor(peer_mem)
363-
_ = ash.aclshmem_finialize()
363+
_ = ash.aclshmem_finalize()
364364

365365

366366
if __name__ == "__main__":

0 commit comments

Comments
 (0)