multi_test.sh

#!/bin/bash
#rm ~/.config/miopen -rf
#rm ~/.cache/miopen -rf
export HIP_VISIBLE_DEVICES=0,1
export MIOPEN_FIND_MODE=3
export HSA_FORCE_FINE_GRAIN_PCIE=1
#export MIOPEN_LOG_LEVEL=4
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