mirror of
https://github.com/microsoft/mscclpp.git
synced 2026-05-11 17:00:22 +00:00
19 lines
722 B
Bash
Executable File
19 lines
722 B
Bash
Executable File
#!/bin/bash
|
|
set -ex
|
|
|
|
# Check if the number of arguments is exactly 1
|
|
if [ "$#" -ne 3 ]; then
|
|
echo "Usage: $0 <hostfile> <nnodes> <ppn>"
|
|
exit 1
|
|
fi
|
|
|
|
HOSTFILE=$1
|
|
NNODES=$2
|
|
PPN=$3
|
|
|
|
parallel-scp -h "$HOSTFILE" -p32 -t1800 -r python/test/executor_test.py /home/azhpcuser/mahdieh/mscclpp/python/test/
|
|
|
|
parallel-scp -h "$HOSTFILE" -p32 -t1800 -r python/mscclpp/default_algos/mscclpp_send_recv.py /home/azhpcuser/mahdieh/mscclpp/python/mscclpp/default_algos/
|
|
|
|
parallel-ssh -h "$HOSTFILE" -p32 -i -t1800 "cd /home/azhpcuser/mahdieh/mscclpp && source mscclpp/bin/activate && python3 python/mscclpp/default_algos/mscclpp_send_recv.py --name send_recv_test --nnodes $NNODES --gpus_per_node $PPN --split_mask 0x3 > test.json "
|