for para in $* do if [[ $para == --profiling* ]];then profiling=${para#*=} fi done
mpirun -np 8 --allow-run-as-root \
train_gpt_567B_1nodes.sh localhost --profiling=$profiling > output.log 2>&1
wait rm -rf CKPT rm -rf mixtral_dataset/my-mixtral_text_document