Mathis/CUDANet
1
0
Fork 0
mirror of https://github.com/lordmathis/CUDANet.git synced 2025-11-06 01:34:22 +00:00
Code Issues Packages Projects Releases Wiki Activity

Add cudaDeviceReset at the end of each test

Browse Source
This commit is contained in:
Matúš Námešný
2024-04-11 19:55:02 +02:00
parent bc86ed1782
commit bf7c961b9e
10 changed files with 93 additions and 10 deletions
Download Patch File Download Diff File Expand all files Collapse all files

13
test/kernels/test_matmul.cu
View File

@@ -45,6 +45,8 @@ TEST(MatMulTest, MatVecMulTest) {
int THREADS_PER_BLOCK = std::max(w, h);
int BLOCKS = 1;
CUDANet::Kernels::clear<<<BLOCKS, h>>>(d_output, h);
CUDANet::Kernels::mat_vec_mul<<<BLOCKS, THREADS_PER_BLOCK, sizeof(float) * w>>>(d_matrix, d_vector, d_output, w, h);
cudaStatus = cudaDeviceSynchronize();
EXPECT_EQ(cudaStatus, cudaSuccess);
@@ -60,6 +62,12 @@ TEST(MatMulTest, MatVecMulTest) {
}
EXPECT_NEAR(sum, output_gpu[i], 1e-5);
}
cudaFree(d_matrix);
cudaFree(d_vector);
cudaFree(d_output);
cudaDeviceReset();
}
TEST(MatMulTest, MaxReduceTest) {
@@ -89,4 +97,9 @@ TEST(MatMulTest, MaxReduceTest) {
EXPECT_EQ(cudaStatus, cudaSuccess);
EXPECT_EQ(output[0], 0.932f);
cudaFree(d_input);
cudaFree(d_output);
cudaDeviceReset();
}