xgboost/tests/distributed/runtests-mpi.sh
2020-11-21 13:38:21 +08:00

14 lines
383 B
Bash
Executable File

#!/bin/bash
rm -f *.model*
export DMLC_SUBMIT_CLUSTER=mpi
submit="timeout 5 python ../../dmlc-core/tracker/dmlc-submit"
echo "====== 1. Basic distributed test with Python ======"
$submit --cluster=local --num-workers=3 python test_basic.py
echo "====== 2. Regression test for issue #3402 ======"
$submit --cluster=local --num-workers=2 --worker-cores=1 python test_issue3402.py