[RUNTIME/CUDA] Allow save exit when driver unload (#122)

c154fcdc · Tianqi Chen · GitHub · f364d563 · c154fcdc
Commit c154fcdc authored May 06, 2017 by Tianqi Chen Committed by GitHub May 06, 2017
Hide whitespace changes
Inline Side-by-side

Showing with 1 additions and 1 deletions

src/runtime/cuda/cuda_common.h
+1 -1

No files found.
--- a/src/runtime/cuda/cuda_common.h
+++ b/src/runtime/cuda/cuda_common.h
@@ -19,7 +19,7 @@ namespace runtime {
 #define CUDA_DRIVER_CALL(x)                                             \
  {                                                                     \
    CUresult result = x;                                                \
-    if (result != CUDA_SUCCESS) {                                       \
+    if (result != CUDA_SUCCESS && result != CUDA_ERROR_DEINITIALIZED) { \
      const char *msg;                                                  \
      cuGetErrorName(result, &msg);                                     \
      LOG(FATAL)                                                        \