diff options
author | Hieu Hoang <hieuhoang@gmail.com> | 2018-02-03 01:05:05 +0300 |
---|---|---|
committer | Hieu Hoang <hieuhoang@gmail.com> | 2018-02-03 01:05:05 +0300 |
commit | 2b134fbf49baf860a9d2cb37529923746c6721d0 (patch) | |
tree | 2914777a081ec5e5917aea008e55880bee2a3f37 | |
parent | c01867dfaec96350925e3fc21d428d982d90cf27 (diff) |
abort if cuda errortopup-batch
-rw-r--r-- | src/amun/gpu/types-gpu.cu | 6 |
1 files changed, 4 insertions, 2 deletions
diff --git a/src/amun/gpu/types-gpu.cu b/src/amun/gpu/types-gpu.cu index 6d569e68..9ad279d4 100644 --- a/src/amun/gpu/types-gpu.cu +++ b/src/amun/gpu/types-gpu.cu @@ -10,14 +10,16 @@ namespace GPU { void HandleError(cudaError_t err, const char *file, int line ) { if (err != cudaSuccess) { std::cerr << "ERROR: " << cudaGetErrorString(err) << " in " << file << " at line " << line << std::endl; - exit( EXIT_FAILURE ); + abort(); + //exit( EXIT_FAILURE ); } } void HandleErrorCublas(cublasStatus_t err, const char *file, int line ) { if (err != CUBLAS_STATUS_SUCCESS) { std::cerr << "CUBLAS ERROR: " << err << " in " << file << " at line " << line << std::endl; - exit( EXIT_FAILURE ); + abort(); + //exit( EXIT_FAILURE ); } } |