Skip to content

Commit c3e3733

Browse files
committed
ROCm fixes
1 parent 15db19a commit c3e3733

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

ggml-cuda.cu

+2-1
Original file line numberDiff line numberDiff line change
@@ -34,6 +34,7 @@
3434
#define cudaEventDisableTiming hipEventDisableTiming
3535
#define cudaEventRecord hipEventRecord
3636
#define cudaEvent_t hipEvent_t
37+
#define cudaEventDestroy hipEventDestroy
3738
#define cudaFree hipFree
3839
#define cudaFreeHost hipHostFree
3940
#define cudaGetDevice hipGetDevice
@@ -56,7 +57,7 @@
5657
#define cudaStreamCreateWithFlags hipStreamCreateWithFlags
5758
#define cudaStreamNonBlocking hipStreamNonBlocking
5859
#define cudaStreamSynchronize hipStreamSynchronize
59-
#define cudaStreamWaitEvent hipStreamWaitEvent
60+
#define cudaStreamWaitEvent(stream, event) hipStreamWaitEvent(stream, event, 0)
6061
#define cudaStream_t hipStream_t
6162
#define cudaSuccess hipSuccess
6263
#else

0 commit comments

Comments
 (0)