microsoft/mscclpp
1
0
Fork 0
mirror of https://github.com/microsoft/mscclpp.git synced 2026-04-20 06:49:29 +00:00
Code Issues Packages Projects Releases Wiki Activity

AlltoAll Test Support (#606)

Browse Source 
Co-authored-by: Binyang Li <binyli@microsoft.com>
This commit is contained in:
Caio Rocha
2025-08-15 16:00:41 -07:00
committed by GitHub
parent 2b40fe37b3
commit 9261b1d278
13 changed files with 778 additions and 23 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
python/test/executor_test.py
View File

@@ -75,8 +75,10 @@ def bench_correctness(
coll = "all_gather"
elif "reducescatter" in collective:
coll = "reduce_scatter"
else:
elif "allreduce" in collective:
coll = "all_reduce"
else:
coll = "all_to_all"
test_data_kernel_name = "test_data_%s_%s" % (coll, dtype_str)
file_dir = os.path.dirname(os.path.abspath(__file__))