Skip to content

GitLab

Commit 9727a645 authored by one's avatar one
Browse files

Update 8-GPU RCCL envs for BW150

parent 8794da26
Hide whitespace changes
Inline Side-by-side
Showing with 6 additions and 0 deletions
projects/rccl-tests/bw150/run-allreduce.sh
View file @ 9727a645
...@@ -23,3 +23,9 @@ NCCL_GRAPH_FILE=graph-allreduce-4r-0123.xml \ ...@@ -23,3 +23,9 @@ NCCL_GRAPH_FILE=graph-allreduce-4r-0123.xml \
HIP_VISIBLE_DEVICES=4,5,6,7 \ HIP_VISIBLE_DEVICES=4,5,6,7 \
NCCL_GRAPH_FILE=graph-allreduce-4r-4567.xml \ NCCL_GRAPH_FILE=graph-allreduce-4r-4567.xml \
./build/all_reduce_perf -b 4 -e 16G -f 2 -w 3 -n 3 -g 4 ./build/all_reduce_perf -b 4 -e 16G -f 2 -w 3 -n 3 -g 4
# GPUs: 0,1,2,3,4,5,6,7
# DTK26 之前使用如下环境变量
NCCL_ALGO=Tree \
NCCL_RINGS="0 1 2 3 4 5 6 7|7 6 5 4 3 2 1 0|0 2 3 1 4 6 7 5|5 7 6 4 1 3 2 0|0 3 1 2 4 7 5 6|6 5 7 4 2 1 3 0" \
./build/all_reduce_perf -b 4 -e 16G -f 2 -w 3 -n 3 -g 8
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment