summaryrefslogtreecommitdiff
diff options
context:
space:
mode:
-rw-r--r--ggml-cuda.cu1
1 files changed, 1 insertions, 0 deletions
diff --git a/ggml-cuda.cu b/ggml-cuda.cu
index 036668bf..61d92d7e 100644
--- a/ggml-cuda.cu
+++ b/ggml-cuda.cu
@@ -80,6 +80,7 @@
#define cudaStreamWaitEvent(stream, event, flags) hipStreamWaitEvent(stream, event, flags)
#define cudaStream_t hipStream_t
#define cudaSuccess hipSuccess
+#define __trap abort
#else
#include <cuda_runtime.h>
#include <cublas_v2.h>