microsoft/mscclpp
1
0
Fork 0
mirror of https://github.com/microsoft/mscclpp.git synced 2026-05-12 01:10:22 +00:00
Code Issues Packages Projects Releases Wiki Activity

add sendrecv correctness check

Browse Source
This commit is contained in:
Ubuntu
2026-04-03 20:01:22 +00:00
parent a4bb8fb4bf
commit 194a79f772
1 changed files with 2 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
python/test/executor_test.py
View File

@@ -70,6 +70,7 @@ def bench_correctness(
):
type_size = cp.dtype(parse_dtype(dtype_str)).itemsize
print("collective: ", collective)
fill_data_kernel_name = "fill_data_%s" % dtype_str
if "allgather" in collective:
coll = "all_gather"
@@ -78,7 +79,7 @@ def bench_correctness(
elif "allreduce" in collective:
coll = "all_reduce"
else:
coll = "all_to_all"
coll = "sendrecv"
test_data_kernel_name = "test_data_%s_%s" % (coll, dtype_str)
file_dir = os.path.dirname(os.path.abspath(__file__))