fix hip build
This commit is contained in:
parent
89bfa1f2ed
commit
aa1e2f8b2f
1 changed files with 2 additions and 1 deletions
|
@ -72,6 +72,7 @@
|
||||||
#define cudaEventCreateWithFlags hipEventCreateWithFlags
|
#define cudaEventCreateWithFlags hipEventCreateWithFlags
|
||||||
#define cudaEventDisableTiming hipEventDisableTiming
|
#define cudaEventDisableTiming hipEventDisableTiming
|
||||||
#define cudaEventRecord hipEventRecord
|
#define cudaEventRecord hipEventRecord
|
||||||
|
#define cudaEventSynchronize hipEventSynchronize
|
||||||
#define cudaEvent_t hipEvent_t
|
#define cudaEvent_t hipEvent_t
|
||||||
#define cudaEventDestroy hipEventDestroy
|
#define cudaEventDestroy hipEventDestroy
|
||||||
#define cudaFree hipFree
|
#define cudaFree hipFree
|
||||||
|
@ -105,6 +106,7 @@
|
||||||
#define cudaStreamCreateWithFlags hipStreamCreateWithFlags
|
#define cudaStreamCreateWithFlags hipStreamCreateWithFlags
|
||||||
#define cudaStreamFireAndForget hipStreamFireAndForget
|
#define cudaStreamFireAndForget hipStreamFireAndForget
|
||||||
#define cudaStreamNonBlocking hipStreamNonBlocking
|
#define cudaStreamNonBlocking hipStreamNonBlocking
|
||||||
|
#define cudaStreamPerThread hipStreamPerThread
|
||||||
#define cudaStreamSynchronize hipStreamSynchronize
|
#define cudaStreamSynchronize hipStreamSynchronize
|
||||||
#define cudaStreamWaitEvent(stream, event, flags) hipStreamWaitEvent(stream, event, flags)
|
#define cudaStreamWaitEvent(stream, event, flags) hipStreamWaitEvent(stream, event, flags)
|
||||||
#define cudaStream_t hipStream_t
|
#define cudaStream_t hipStream_t
|
||||||
|
@ -11531,7 +11533,6 @@ static void ggml_backend_cuda_event_record(ggml_backend_event_t event) {
|
||||||
ggml_backend_cuda_context * cuda_ctx = (ggml_backend_cuda_context *)event->backend->context;
|
ggml_backend_cuda_context * cuda_ctx = (ggml_backend_cuda_context *)event->backend->context;
|
||||||
|
|
||||||
CUDA_CHECK(cudaEventRecord((cudaEvent_t)event->context, g_cudaStreams[cuda_ctx->device][0]));
|
CUDA_CHECK(cudaEventRecord((cudaEvent_t)event->context, g_cudaStreams[cuda_ctx->device][0]));
|
||||||
cudaError_t err = cudaEventQuery((cudaEvent_t)event->context);
|
|
||||||
}
|
}
|
||||||
|
|
||||||
static void ggml_backend_cuda_event_wait(ggml_backend_t backend, ggml_backend_event_t event) {
|
static void ggml_backend_cuda_event_wait(ggml_backend_t backend, ggml_backend_event_t event) {
|
||||||
|
|
Loading…
Add table
Add a link
Reference in a new issue