2020-11-10 15:44:10 -07:00
|
|
|
#!/bin/bash
|
|
|
|
|
RCCL_INSTALL=../../build/release
|
|
|
|
|
EXE=$PWD/HelloRccl
|
|
|
|
|
LDPATH=$LD_LIBRARY_PATH:$RCCL_INSTALL
|
|
|
|
|
|
2025-03-20 18:10:01 -06:00
|
|
|
echo "Single process:"
|
2020-11-10 15:44:10 -07:00
|
|
|
NCCL_DEBUG=INFO LD_LIBRARY_PATH=$LDPATH $EXE 4
|
|
|
|
|
|
2025-03-20 18:10:01 -06:00
|
|
|
echo "Multi-process:"
|
2020-11-10 15:44:10 -07:00
|
|
|
NCCL_COMM_ID=$HOSTNAME:12345 LD_LIBRARY_PATH=$LDPATH $EXE 4 0 &
|
|
|
|
|
NCCL_COMM_ID=$HOSTNAME:12345 LD_LIBRARY_PATH=$LDPATH $EXE 4 1 &
|
|
|
|
|
NCCL_COMM_ID=$HOSTNAME:12345 LD_LIBRARY_PATH=$LDPATH $EXE 4 2 &
|
|
|
|
|
NCCL_COMM_ID=$HOSTNAME:12345 LD_LIBRARY_PATH=$LDPATH $EXE 4 3
|
