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

show scale in output

Browse Source
This commit is contained in:
Ubuntu
2026-03-17 20:43:32 +00:00
parent 27fbddb707
commit d07a1ba28c
1 changed files with 3 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
python/test/executor_test.py
View File

@@ -327,8 +327,9 @@ def main(
npkit.dump(npkit_dump_dir)
npkit.shutdown()
mscclpp_group.barrier()
print("size= ", size, "nelem= ", nelem)
# Print header once
print(f"{'NRanks':>8} {'Message Size (B)':>18} {'BW (GB/s)':>12} {'Latency (us)':>14} {'Packet Type':>12}")
print(f"{nranks:8d} {msg_size:18d} {bw:12.2f} {latency:14.2f} {str(packet_type):>12}")
# Sentinel fill: choose something unlikely in your pattern
result_buf.fill(cp.float16(123.0))