Add cudaDeviceReset at the end of each test

2025-11-06 09:44:28 +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_conv2d.cu
+++ b/test/layers/test_conv2d.cu
@@ -47,6 +47,7 @@ class Conv2dTest : public ::testing::Test {
    void commonTestTeardown(float* d_input) {
        // Free device memory
        cudaFree(d_input);
+        cudaDeviceReset();
    }

    cudaError_t cudaStatus;