@@ -35,6 +35,7 @@ void throw_fork_cuda_exc() { | |||||
// call chain: | // call chain: | ||||
// python -> fork() -> pthread_atfork -> CudaCheckOnFork -> | // python -> fork() -> pthread_atfork -> CudaCheckOnFork -> | ||||
// ForkAfterCudaError::throw_ | // ForkAfterCudaError::throw_ | ||||
GILManager a; | |||||
mgb_log_warn("try to raise python exception for fork after cuda"); | mgb_log_warn("try to raise python exception for fork after cuda"); | ||||
PyErr_SetString(PyExc_SystemError, "fork after cuda has been initialized"); | PyErr_SetString(PyExc_SystemError, "fork after cuda has been initialized"); | ||||
} | } | ||||
@@ -1,9 +1,13 @@ | |||||
#!/bin/bash | |||||
#!/bin/bash -e | |||||
cd $(dirname "${BASH_SOURCE[0]}")/.. | |||||
pytest -m 'not internet' \ | |||||
--ignore test/pytorch_comparison \ | |||||
--ignore test/integration/manual \ | |||||
--ignore megengine/docs \ | |||||
megengine test | |||||
pushd $(dirname "${BASH_SOURCE[0]}")/.. >/dev/null | |||||
pytest -xv -m 'not internet'\ | |||||
--ignore test/unit/module/test_pytorch.py \ | |||||
--ignore test/pytorch_comparison \ | |||||
--ignore test/unit/hub/test_hub.py \ | |||||
--ignore test/unit/data \ | |||||
--ignore test/integration/manual \ | |||||
--ignore megengine/docs/ \ | |||||
--ignore megengine/module/pytorch \ | |||||
megengine test | |||||
popd >/dev/null |