mirror of
https://github.com/lordmathis/CUDANet.git
synced 2025-11-05 17:34:21 +00:00
Remove cudaDeviceReset calls from tests
This commit is contained in:
@@ -45,5 +45,5 @@ TEST(ActivationFunctionsTest, SigmoidSanityCheck) {
|
||||
cudaFree(d_input);
|
||||
cudaFree(d_output);
|
||||
|
||||
cudaDeviceReset();
|
||||
|
||||
}
|
||||
|
||||
@@ -68,7 +68,7 @@ TEST(MatMulTest, MatVecMulTest) {
|
||||
cudaFree(d_vector);
|
||||
cudaFree(d_output);
|
||||
|
||||
cudaDeviceReset();
|
||||
|
||||
}
|
||||
|
||||
TEST(MatMulTest, MaxReduceTest) {
|
||||
@@ -113,7 +113,7 @@ TEST(MatMulTest, MaxReduceTest) {
|
||||
cudaFree(d_input);
|
||||
cudaFree(d_output);
|
||||
|
||||
cudaDeviceReset();
|
||||
|
||||
}
|
||||
|
||||
TEST(MatMulTest, VecExpTest) {
|
||||
@@ -157,7 +157,7 @@ TEST(MatMulTest, VecExpTest) {
|
||||
cudaFree(d_input);
|
||||
cudaFree(d_output);
|
||||
|
||||
cudaDeviceReset();
|
||||
|
||||
}
|
||||
|
||||
TEST(MatMulTest, SumReduceTest) {
|
||||
@@ -210,5 +210,5 @@ TEST(MatMulTest, SumReduceTest) {
|
||||
cudaFree(d_input);
|
||||
cudaFree(d_sum);
|
||||
|
||||
cudaDeviceReset();
|
||||
|
||||
}
|
||||
@@ -38,10 +38,7 @@ TEST(ActivationTest, SoftmaxTest1) {
|
||||
|
||||
EXPECT_NEAR(sum, 1.0f, 1e-5f);
|
||||
|
||||
cudaFree(d_input);
|
||||
cudaDeviceReset();
|
||||
|
||||
cudaStatus = cudaGetLastError();
|
||||
cudaStatus = cudaFree(d_input);
|
||||
EXPECT_EQ(cudaStatus, cudaSuccess);
|
||||
}
|
||||
|
||||
@@ -82,6 +79,6 @@ TEST(ActivationTest, SoftmaxTest2) {
|
||||
EXPECT_NEAR(sum, 1.0f, 1e-5f);
|
||||
|
||||
// Cleanup
|
||||
cudaFree(d_input);
|
||||
cudaDeviceReset();
|
||||
cudaStatus = cudaFree(d_input);
|
||||
EXPECT_EQ(cudaStatus, cudaSuccess);
|
||||
}
|
||||
@@ -67,6 +67,4 @@ TEST(AvgPoolingLayerTest, AvgPoolForwardTest) {
|
||||
|
||||
cudaFree(d_input);
|
||||
cudaFree(d_output);
|
||||
|
||||
cudaDeviceReset();
|
||||
}
|
||||
|
||||
@@ -34,6 +34,4 @@ TEST(ConcatLayerTest, Init) {
|
||||
EXPECT_EQ(output[i + 5], inputB[i]);
|
||||
}
|
||||
cudaFree(d_output);
|
||||
|
||||
cudaDeviceReset();
|
||||
}
|
||||
@@ -47,7 +47,7 @@ class Conv2dTest : public ::testing::Test {
|
||||
void commonTestTeardown(float* d_input) {
|
||||
// Free device memory
|
||||
cudaFree(d_input);
|
||||
cudaDeviceReset();
|
||||
|
||||
}
|
||||
|
||||
cudaError_t cudaStatus;
|
||||
|
||||
@@ -41,7 +41,7 @@ class DenseLayerTest : public ::testing::Test {
|
||||
void commonTestTeardown(float* d_input) {
|
||||
// Free device memory
|
||||
cudaFree(d_input);
|
||||
cudaDeviceReset();
|
||||
|
||||
}
|
||||
|
||||
cudaError_t cudaStatus;
|
||||
|
||||
@@ -15,5 +15,5 @@ TEST(InputLayerTest, InputForward) {
|
||||
EXPECT_EQ(cudaStatus, cudaSuccess);
|
||||
EXPECT_EQ(input, output);
|
||||
|
||||
cudaDeviceReset();
|
||||
|
||||
}
|
||||
@@ -68,5 +68,5 @@ TEST(MaxPoolingLayerTest, MaxPoolForwardTest) {
|
||||
cudaFree(d_input);
|
||||
cudaFree(d_output);
|
||||
|
||||
cudaDeviceReset();
|
||||
|
||||
}
|
||||
|
||||
@@ -23,5 +23,5 @@ TEST(OutputLayerTest, OutputForward) {
|
||||
}
|
||||
|
||||
cudaFree(d_input);
|
||||
cudaDeviceReset();
|
||||
|
||||
}
|
||||
Reference in New Issue
Block a user