Welcome to mirror list, hosted at ThFree Co, Russian Federation.

github.com/marian-nmt/marian.git - Unnamed repository; edit this file 'description' to name the repository.
summaryrefslogtreecommitdiff
diff options
context:
space:
mode:
authorHieu Hoang <hieuhoang@gmail.com>2018-02-03 01:05:05 +0300
committerHieu Hoang <hieuhoang@gmail.com>2018-02-03 01:05:05 +0300
commit2b134fbf49baf860a9d2cb37529923746c6721d0 (patch)
tree2914777a081ec5e5917aea008e55880bee2a3f37
parentc01867dfaec96350925e3fc21d428d982d90cf27 (diff)
abort if cuda errortopup-batch
-rw-r--r--src/amun/gpu/types-gpu.cu6
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 );
}
}