Add cudaDeviceReset at the end of each test

This commit is contained in:
2024-04-11 19:55:02 +02:00
parent bc86ed1782
commit bf7c961b9e
10 changed files with 93 additions and 10 deletions

View File

@@ -14,4 +14,6 @@ TEST(InputLayerTest, InputForward) {
);
EXPECT_EQ(cudaStatus, cudaSuccess);
EXPECT_EQ(input, output);
cudaDeviceReset();
}