From: Mahekk Shaikh <redacted>
Date: Sun, 30 Nov 2025 00:16:28 +0000 (-0500)
Subject: cuda : add error checking for cudaMemcpyAsync in argsort (llama/17599)
X-Git-Tag: upstream/1.8.3~214
X-Git-Url: https://git.djapps.eu/?a=commitdiff_plain;h=2e4a7a21faa5198e2501fb499c8449c750e52f02;p=pkg%2Fggml%2Fsources%2Fwhisper.cpp

cuda : add error checking for cudaMemcpyAsync in argsort (llama/17599)

* cuda : add error checking for cudaMemcpyAsync in argsort (llama/12836)

* fix indentation
---

diff --git a/ggml/src/ggml-cuda/argsort.cu b/ggml/src/ggml-cuda/argsort.cu
index 3722cf3a..da9652c3 100644
--- a/ggml/src/ggml-cuda/argsort.cu
+++ b/ggml/src/ggml-cuda/argsort.cu
@@ -44,7 +44,7 @@ static void argsort_f32_i32_cuda_cub(ggml_cuda_pool & pool,
     const dim3 offset_grid((nrows + block_size - 1) / block_size);
     init_offsets<<<offset_grid, block_size, 0, stream>>>(d_offsets, ncols, nrows);
 
-    cudaMemcpyAsync(temp_keys, x, ncols * nrows * sizeof(float), cudaMemcpyDeviceToDevice, stream);
+    CUDA_CHECK(cudaMemcpyAsync(temp_keys, x, ncols * nrows * sizeof(float), cudaMemcpyDeviceToDevice, stream));
 
     size_t temp_storage_bytes = 0;