Catch dmlc::Error. (#3751)

Fix #3643.
dmlc · Oct 4, 2018 · c6b5df6 · c6b5df6
1 parent efc4f85
commit c6b5df6
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/src/common/common.cu b/src/common/common.cu
@@ -11,7 +11,7 @@ int AllVisibleImpl::AllVisible() {
     // When compiled with CUDA but running on CPU only device,
     // cudaGetDeviceCount will fail.
     dh::safe_cuda(cudaGetDeviceCount(&n_visgpus));
-  } catch(const thrust::system::system_error& err) {
+  } catch(const dmlc::Error &except) {
     return 0;
   }
   return n_visgpus;

diff --git a/src/common/device_helpers.cuh b/src/common/device_helpers.cuh
@@ -956,7 +956,7 @@ class SaveCudaContext {
     // cudaGetDevice will fail.
     try {
       safe_cuda(cudaGetDevice(&saved_device_));
-    } catch (const thrust::system::system_error & err) {
+    } catch (const dmlc::Error &except) {
       saved_device_ = -1;
     }
     func();