multi_test.sh

#!/bin/bash
export HIP_VISIBLE_DEVICES=0,1
export MIOPEN_FIND_MODE=3
export HSA_FORCE_FINE_GRAIN_PCIE=1
my_config=$1
numactl --cpunodebind=2 --membind=2 python3 -m torch.distributed.launch --nproc_per_node=2 --master_port=29500   tools/train.py $my_config --launcher pytorch