Remove extra cudaFree

2025-11-06 09:44:28 +00:00 · 2024-04-22 19:05:02 +02:00
parent a32c737785
commit fbf6c44bdd
2 changed files with 4 additions and 5 deletions
--- 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);
 }