conjugateGradient: Ensure allocated memory is freed (#336)

Fixes #202
2026-10-25 22:28:17 +08:00 · 2025-02-21 11:20:53 -06:00
parent 3bc08136ff
commit 29fb758e62
1 changed files with 4 additions and 0 deletions
--- a/Samples/4_CUDA_Libraries/conjugateGradient/main.cpp
+++ b/Samples/4_CUDA_Libraries/conjugateGradient/main.cpp
@@ -231,6 +231,10 @@ int main(int argc, char **argv) {
    }
  }

+  if (buffer) {
+    checkCudaErrors(cudaFree(buffer));
+  }
+  
  cusparseDestroy(cusparseHandle);
  cublasDestroy(cublasHandle);
  if (matA) {