ggml-cuda: Fix HIP build by adding define for __trap (ggerganov#4569)

Regression of 1398823 HIP doesn't have trap, only abort
heiner · Dec 21, 2023 · 562cf22 · 562cf22
1 parent 8fe03ff
commit 562cf22
Showing 1 changed file with 1 addition and 0 deletions.
diff --git 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>