From 00425e2ed1d1ec35976710f81a9337c7b2d34d96 Mon Sep 17 00:00:00 2001 From: Mahekk Shaikh <118063190+Mahekk357@users.noreply.github.com> Date: Sat, 29 Nov 2025 19:16:28 -0500 Subject: [PATCH] cuda : add error checking for cudaMemcpyAsync in argsort (#17599) * cuda : add error checking for cudaMemcpyAsync in argsort (#12836) * fix indentation --- ggml/src/ggml-cuda/argsort.cu | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/ggml/src/ggml-cuda/argsort.cu b/ggml/src/ggml-cuda/argsort.cu index 3722cf3ab2..da9652c3be 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<<>>(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;