提交 9edce7e2 编写于 作者: M Megvii Engine Team

ci(python_test): run python test with cuda

GitOrigin-RevId: bc51c2e4d554d46cb4596c4e1629f7de8934e6c4
上级 5fe7cd14
......@@ -35,6 +35,7 @@ void throw_fork_cuda_exc() {
// call chain:
// python -> fork() -> pthread_atfork -> CudaCheckOnFork ->
// ForkAfterCudaError::throw_
GILManager a;
mgb_log_warn("try to raise python exception for fork after cuda");
PyErr_SetString(PyExc_SystemError, "fork after cuda has been initialized");
}
......
#!/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
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册