From: Erik Scholz <redacted>
Date: Fri, 5 Jan 2024 15:00:00 +0000 (+0100)
Subject: fix : cuda order of synchronization when setting a buffer (#679)
X-Git-Tag: upstream/0.0.1642~1121
X-Git-Url: https://git.djapps.eu/?a=commitdiff_plain;h=3eace58911ea8d2cf35defdc59848d99b91a57f5;p=pkg%2Fggml%2Fsources%2Fggml

fix : cuda order of synchronization when setting a buffer (#679)

* fix : cuda order of synchronization when setting a buffer

* also sync before memcpy

---------

Co-authored-by: slaren <redacted>
---

diff --git a/src/ggml-cuda.cu b/src/ggml-cuda.cu
index 10c21615..7578d21c 100644
--- a/src/ggml-cuda.cu
+++ b/src/ggml-cuda.cu
@@ -9689,8 +9689,8 @@ static void ggml_backend_cuda_buffer_set_tensor(ggml_backend_buffer_t buffer, gg
 
     ggml_cuda_set_device(ctx->device);
     CUDA_CHECK(cudaDeviceSynchronize());
-
     CUDA_CHECK(cudaMemcpy((char *)tensor->data + offset, data, size, cudaMemcpyHostToDevice));
+    CUDA_CHECK(cudaDeviceSynchronize());
 }
 
 static void ggml_backend_cuda_buffer_get_tensor(ggml_backend_buffer_t buffer, const ggml_tensor * tensor, void * data, size_t offset, size_t size) {