Skip to content

Commit c3d4ead

Browse files
committed
added missing CUDA_CHECKs
1 parent df4719e commit c3d4ead

File tree

1 file changed

+2
-3
lines changed

1 file changed

+2
-3
lines changed

ggml-cuda.cu

Lines changed: 2 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -2440,12 +2440,11 @@ GGML_CALL static enum ggml_status ggml_backend_cuda_graph_compute(ggml_backend_t
24402440
// kernel parameters which need updated in the graph for each token
24412441
void* ggmlCudaCpyFn = nullptr;
24422442

2443-
24442443
if(cudaGraph.count==0){
24452444
cudaDeviceProp prop;
24462445
int device;
2447-
cudaGetDevice(&device);
2448-
cudaGetDeviceProperties(&prop, device);
2446+
CUDA_CHECK(cudaGetDevice(&device));
2447+
CUDA_CHECK(cudaGetDeviceProperties(&prop, device));
24492448
if (prop.major < 8){
24502449
cudaGraph.disableDueToGpuArch=true;
24512450
}

0 commit comments

Comments
 (0)