Browse Source

ci(pytest): use common method install pip package

GitOrigin-RevId: 87162b8b56
tags/v0.3.2
Megvii Engine Team 5 years ago
parent
commit
646829b802
2 changed files with 3 additions and 1 deletions
  1. +3
    -0
      python_module/setup.py
  2. +0
    -1
      python_module/src/cpp/function_replace.cpp

+ 3
- 0
python_module/setup.py View File

@@ -87,6 +87,9 @@ setup_kwargs = dict(
'data': [
'scipy',
],
'ci': [
'pytest==5.3.0',
],
},
cmdclass={'build_ext': build_ext},
)


+ 0
- 1
python_module/src/cpp/function_replace.cpp View File

@@ -35,7 +35,6 @@ 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");
}


Loading…
Cancel
Save