mirror of
https://github.com/microsoft/mscclpp.git
synced 2026-05-12 01:10:22 +00:00
Port python tests to mscclpp. Please run `mpirun -tag-output -np 8 pytest ./python/test/test_mscclpp.py -x` to start pytest --------- Co-authored-by: Saeed Maleki <saemal@microsoft.com> Co-authored-by: Changho Hwang <changhohwang@microsoft.com> Co-authored-by: Saeed Maleki <30272783+saeedmaleki@users.noreply.github.com>
8 lines
188 B
Bash
8 lines
188 B
Bash
#!/bin/bash
|
|
if [[ $OMPI_COMM_WORLD_RANK == 0 ]]
|
|
then
|
|
pytest /root/mscclpp/python/test/test_mscclpp.py -x -v
|
|
else
|
|
pytest /root/mscclpp/python/test/test_mscclpp.py -x 2>&1 >/dev/null
|
|
fi
|