2b8184808d
* Initial support for clique-based kernels
17 γραμμές
783 B
Bash
Εκτελέσιμο Αρχείο
17 γραμμές
783 B
Bash
Εκτελέσιμο Αρχείο
#!/bin/bash
|
|
RCCL_INSTALL=../../build/release
|
|
EXE=$PWD/HelloRccl
|
|
LDPATH=$LD_LIBRARY_PATH:$RCCL_INSTALL
|
|
|
|
echo "With clique-based kernels:"
|
|
RCCL_ENABLE_CLIQUE=1 NCCL_COMM_ID=$HOSTNAME:12345 LD_LIBRARY_PATH=$LDPATH $EXE 4 0 0 &
|
|
RCCL_ENABLE_CLIQUE=1 NCCL_COMM_ID=$HOSTNAME:12345 LD_LIBRARY_PATH=$LDPATH $EXE 4 1 1 &
|
|
RCCL_ENABLE_CLIQUE=1 NCCL_COMM_ID=$HOSTNAME:12345 LD_LIBRARY_PATH=$LDPATH $EXE 4 2 2 &
|
|
RCCL_ENABLE_CLIQUE=1 NCCL_COMM_ID=$HOSTNAME:12345 LD_LIBRARY_PATH=$LDPATH $EXE 4 3 3
|
|
|
|
echo "Without clique-based kernels:"
|
|
NCCL_COMM_ID=$HOSTNAME:12345 LD_LIBRARY_PATH=$LDPATH $EXE 4 0 0 &
|
|
NCCL_COMM_ID=$HOSTNAME:12345 LD_LIBRARY_PATH=$LDPATH $EXE 4 1 1 &
|
|
NCCL_COMM_ID=$HOSTNAME:12345 LD_LIBRARY_PATH=$LDPATH $EXE 4 2 2 &
|
|
NCCL_COMM_ID=$HOSTNAME:12345 LD_LIBRARY_PATH=$LDPATH $EXE 4 3 3
|