Mathis/CUDANet
1
0
Fork 0
mirror of https://github.com/lordmathis/CUDANet.git synced 2025-11-05 17:34:21 +00:00
Code Issues Packages Projects Releases Wiki Activity

Remove extra cudaFree

Browse Source
This commit is contained in:
Matúš Námešný
2024-04-22 19:05:02 +02:00
parent a32c737785
commit fbf6c44bdd
2 changed files with 4 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
test/layers/test_avg_pooling.cu
View File

@@ -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);
}

6
test/layers/test_max_pooling.cu
View File

@@ -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);
}