mirror of
https://github.com/microsoft/mscclpp.git
synced 2026-05-12 09:17:06 +00:00
add barrier
This commit is contained in:
@@ -230,6 +230,8 @@ def main(
|
||||
# Count how many elements changed
|
||||
changed = cp.count_nonzero(result_buf != cp.float16(123.0)).item()
|
||||
print("changed elements:", changed, "out of", result_buf.size)
|
||||
cp.cuda.runtime.deviceSynchronize()
|
||||
mscclpp_group.barrier()
|
||||
|
||||
bench_correctness(
|
||||
collective,
|
||||
|
||||
Reference in New Issue
Block a user