mirror of
https://github.com/lordmathis/CUDANet.git
synced 2025-11-06 09:44:28 +00:00
Add cudaDeviceReset at the end of each test
This commit is contained in:
@@ -14,4 +14,6 @@ TEST(InputLayerTest, InputForward) {
|
||||
);
|
||||
EXPECT_EQ(cudaStatus, cudaSuccess);
|
||||
EXPECT_EQ(input, output);
|
||||
|
||||
cudaDeviceReset();
|
||||
}
|
||||
Reference in New Issue
Block a user