From 29fb758e628dcc89d2b8429af0ca7df841f22ff2 Mon Sep 17 00:00:00 2001
From: Jonathan Bentz <jnbntz@users.noreply.github.com>
Date: Fri, 21 Feb 2025 11:20:53 -0600
Subject: [PATCH] conjugateGradient: Ensure allocated memory is freed (#336)

Fixes #202
---
 Samples/4_CUDA_Libraries/conjugateGradient/main.cpp | 4 ++++
 1 file changed, 4 insertions(+)

diff --git a/Samples/4_CUDA_Libraries/conjugateGradient/main.cpp b/Samples/4_CUDA_Libraries/conjugateGradient/main.cpp
index a37d3043..50d9470f 100644
--- 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) {