Hendrik/xgboost
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Catch dmlc::Error. (#3751)

Browse Source 
Fix #3643.
This commit is contained in:
trivialfis 2018-10-04 16:51:38 +13:00 committed by Rory Mitchell
parent efc4f85505
commit c6b5df67f6
2 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
src/common/common.cu
View File

@ -11,7 +11,7 @@ int AllVisibleImpl::AllVisible() {
// When compiled with CUDA but running on CPU only device, // When compiled with CUDA but running on CPU only device,
// cudaGetDeviceCount will fail. // cudaGetDeviceCount will fail.
dh::safe_cuda(cudaGetDeviceCount(&n_visgpus)); dh::safe_cuda(cudaGetDeviceCount(&n_visgpus));
} catch(const thrust::system::system_error& err) { } catch(const dmlc::Error &except) {
return 0; return 0;
} }
return n_visgpus; return n_visgpus;

2
src/common/device_helpers.cuh
View File

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