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

Fix more cuda memcheck errors

Browse Source
This commit is contained in:
Matúš Námešný
2024-04-21 22:56:55 +02:00
parent 58af95eb25
commit 26cea9b12c
3 changed files with 25 additions and 15 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
test/layers/test_avg_pooling.cu
View File

@@ -66,5 +66,4 @@ TEST(AvgPoolingLayerTest, AvgPoolForwardTest) {
}
cudaFree(d_input);
cudaFree(d_output);
}

29
test/layers/test_concat.cu
View File

@@ -4,6 +4,9 @@
#include <vector>
TEST(ConcatLayerTest, Init) {
cudaError_t cudaStatus;
std::vector<float> inputA = {0.573f, 0.619f, 0.732f, 0.055f, 0.243f};
std::vector<float> inputB = {0.123f, 0.321f, 0.456f, 0.789f, 0.654f, 0.123f};
@@ -11,27 +14,41 @@ TEST(ConcatLayerTest, Init) {
float* d_inputA;
float* d_inputB;
cudaMalloc((void**)&d_inputA, sizeof(float) * 5);
cudaMalloc((void**)&d_inputB, sizeof(float) * 6);
cudaMemcpy(
cudaStatus = cudaMalloc((void**)&d_inputA, sizeof(float) * 5);
EXPECT_EQ(cudaStatus, cudaSuccess);
cudaStatus = cudaMalloc((void**)&d_inputB, sizeof(float) * 6);
EXPECT_EQ(cudaStatus, cudaSuccess);
cudaStatus = cudaMemcpy(
d_inputA, inputA.data(), sizeof(float) * 5, cudaMemcpyHostToDevice
);
cudaMemcpy(
EXPECT_EQ(cudaStatus, cudaSuccess);
cudaStatus = cudaMemcpy(
d_inputB, inputB.data(), sizeof(float) * 6, cudaMemcpyHostToDevice
);
EXPECT_EQ(cudaStatus, cudaSuccess);
float* d_output = concat.forward(d_inputA, d_inputB);
std::vector<float> output(11);
cudaMemcpy(
cudaStatus = cudaMemcpy(
output.data(), d_output, sizeof(float) * 11, cudaMemcpyDeviceToHost
);
EXPECT_EQ(cudaStatus, cudaSuccess);
for (int i = 0; i < 5; ++i) {
EXPECT_EQ(output[i], inputA[i]);
}
for (int i = 0; i < 6; ++i) {
EXPECT_EQ(output[i + 5], inputB[i]);
}
cudaFree(d_output);
cudaStatus = cudaFree(d_inputA);
EXPECT_EQ(cudaStatus, cudaSuccess);
cudaStatus = cudaFree(d_inputB);
EXPECT_EQ(cudaStatus, cudaSuccess);
}