Files
mscclpp/test/deploy/pytest.sh
Binyang2014 097aa8843a Fix pytest unstable issue. (#170)
- remove `#include <cstdint>` from `poll.hpp`. To make it only contains
device-side code
- Fix compilation issue, which will cause pytest fail randomly. Reuse
the compiled result for same kernel with different arguments
2023-09-06 17:09:04 -07:00

10 lines
196 B
Bash

#!/bin/bash
set -e
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