Skip to content

Commit 2fe9927

Browse files
committed
rocm fixes
1 parent efe7560 commit 2fe9927

File tree

3 files changed

+6
-4
lines changed

3 files changed

+6
-4
lines changed

ggml-cuda.cu

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -33,6 +33,7 @@
3333
#define cudaEventDisableTiming hipEventDisableTiming
3434
#define cudaEventRecord hipEventRecord
3535
#define cudaEvent_t hipEvent_t
36+
#define cudaEventDestroy hipEventDestroy
3637
#define cudaFree hipFree
3738
#define cudaFreeHost hipHostFree
3839
#define cudaGetDevice hipGetDevice
@@ -55,7 +56,7 @@
5556
#define cudaStreamCreateWithFlags hipStreamCreateWithFlags
5657
#define cudaStreamNonBlocking hipStreamNonBlocking
5758
#define cudaStreamSynchronize hipStreamSynchronize
58-
#define cudaStreamWaitEvent hipStreamWaitEvent
59+
#define cudaStreamWaitEvent(stream, event) hipStreamWaitEvent(stream, event, 0)
5960
#define cudaStream_t hipStream_t
6061
#define cudaSuccess hipSuccess
6162
#else

otherarch/ggml_v2-cuda-legacy.cu

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -32,6 +32,7 @@
3232
#define cudaEventDisableTiming hipEventDisableTiming
3333
#define cudaEventRecord hipEventRecord
3434
#define cudaEvent_t hipEvent_t
35+
#define cudaEventDestroy hipEventDestroy
3536
#define cudaFree hipFree
3637
#define cudaFreeHost hipHostFree
3738
#define cudaGetDevice hipGetDevice
@@ -54,7 +55,7 @@
5455
#define cudaStreamCreateWithFlags hipStreamCreateWithFlags
5556
#define cudaStreamNonBlocking hipStreamNonBlocking
5657
#define cudaStreamSynchronize hipStreamSynchronize
57-
#define cudaStreamWaitEvent hipStreamWaitEvent
58+
#define cudaStreamWaitEvent(stream, event) hipStreamWaitEvent(stream, event, 0)
5859
#define cudaStream_t hipStream_t
5960
#define cudaSuccess hipSuccess
6061
#else

otherarch/ggml_v2-cuda.cu

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -32,6 +32,7 @@
3232
#define cudaEventDisableTiming hipEventDisableTiming
3333
#define cudaEventRecord hipEventRecord
3434
#define cudaEvent_t hipEvent_t
35+
#define cudaEventDestroy hipEventDestroy
3536
#define cudaFree hipFree
3637
#define cudaFreeHost hipHostFree
3738
#define cudaGetDevice hipGetDevice
@@ -54,14 +55,13 @@
5455
#define cudaStreamCreateWithFlags hipStreamCreateWithFlags
5556
#define cudaStreamNonBlocking hipStreamNonBlocking
5657
#define cudaStreamSynchronize hipStreamSynchronize
57-
#define cudaStreamWaitEvent hipStreamWaitEvent
58+
#define cudaStreamWaitEvent(stream, event) hipStreamWaitEvent(stream, event, 0)
5859
#define cudaStream_t hipStream_t
5960
#define cudaSuccess hipSuccess
6061
#else
6162
#include <cuda_runtime.h>
6263
#include <cublas_v2.h>
6364
#include <cuda_fp16.h>
64-
6565
#endif
6666

6767
#include "ggml_v2-cuda.h"

0 commit comments

Comments
 (0)