NVIDIA / nccl-tests

NCCL Tests
BSD 3-Clause "New" or "Revised" License
775 stars 226 forks source link

improve parsing of stepbytes (increment size) argument #226

Closed StefanoSalsano closed 6 days ago

StefanoSalsano commented 1 month ago

fixes issue #225 stepbytes (increment size) argument does not support 1M notation

if you run this command .build/all_reduce_perf -b 64M -e 70M -i 2M -g 1

the increment is not parsed as 2M but as 2 bytes, which is quite unexpected

AddyLaddy commented 6 days ago

Thanks!