From fbf6c44bddb76ef9ddae833185104ce151922814 Mon Sep 17 00:00:00 2001
From: LordMathis <matus@namesny.com>
Date: Mon, 22 Apr 2024 19:05:02 +0200
Subject: [PATCH] Remove extra cudaFree

---
 test/layers/test_avg_pooling.cu | 3 ++-
 test/layers/test_max_pooling.cu | 6 ++----
 2 files changed, 4 insertions(+), 5 deletions(-)

diff --git a/test/layers/test_avg_pooling.cu b/test/layers/test_avg_pooling.cu
index 7a13651..30ad3ec 100644
--- a/test/layers/test_avg_pooling.cu
+++ b/test/layers/test_avg_pooling.cu
@@ -65,5 +65,6 @@ TEST(AvgPoolingLayerTest, AvgPoolForwardTest) {
         EXPECT_NEAR(expected[i], output[i], 1e-4);
     }
 
-    cudaFree(d_input);
+    cudaStatus = cudaFree(d_input);
+    EXPECT_EQ(cudaStatus, cudaSuccess);
 }
diff --git a/test/layers/test_max_pooling.cu b/test/layers/test_max_pooling.cu
index 703b82b..8cd6211 100644
--- a/test/layers/test_max_pooling.cu
+++ b/test/layers/test_max_pooling.cu
@@ -65,8 +65,6 @@ TEST(MaxPoolingLayerTest, MaxPoolForwardTest) {
         EXPECT_FLOAT_EQ(expected[i], output[i]);
     }
 
-    cudaFree(d_input);
-    cudaFree(d_output);
-
-    
+    cudaStatus = cudaFree(d_input);
+    EXPECT_EQ(cudaStatus, cudaSuccess);
 }