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

Regression of 1398823922
HIP doesn't have trap, only abort
This commit is contained in:
arlo-phoenix 2023-12-21 20:13:25 +01:00 committed by GitHub
parent 8fe03ffdda
commit 562cf222b5
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

View File

@ -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>