Add cudaDeviceReset at the end of each test

2025-11-06 01:34:22 +00:00 · 2024-04-11 19:55:02 +02:00
parent bc86ed1782
commit bf7c961b9e
10 changed files with 93 additions and 10 deletions
--- a/test/layers/test_concat.cu
+++ b/test/layers/test_concat.cu
@@ -34,4 +34,6 @@ TEST(ConcatLayerTest, Init) {
        EXPECT_EQ(output[i + 5], inputB[i]);
    }
    cudaFree(d_output);
+
+    cudaDeviceReset();
 }