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

Add documentation comments

Browse Source
This commit is contained in:
Matúš Námešný
2024-03-12 21:50:06 +01:00
parent 708164e4d0
commit 7157a27e56
9 changed files with 255 additions and 19 deletions
Download Patch File Download Diff File Expand all files Collapse all files

14
test/layers/test_conv2d.cu
View File

@@ -80,7 +80,7 @@ TEST_F(Conv2dTest, SimpleTest) {
);
int outputSize = (inputSize - kernelSize) / stride + 1;
EXPECT_EQ(outputSize, conv2d.outputSize);
EXPECT_EQ(outputSize, conv2d.getOutputSize());
d_output = conv2d.forward(d_input);
@@ -168,16 +168,16 @@ TEST_F(Conv2dTest, PaddedTest) {
activation, input, kernels.data(), d_input
);
EXPECT_EQ(inputSize, conv2d.outputSize);
EXPECT_EQ(inputSize, conv2d.getOutputSize());
d_output = conv2d.forward(d_input);
std::vector<float> output(
conv2d.outputSize * conv2d.outputSize * numFilters
conv2d.getOutputSize() * conv2d.getOutputSize() * numFilters
);
cudaMemcpy(
output.data(), d_output,
sizeof(float) * conv2d.outputSize * conv2d.outputSize * numFilters,
sizeof(float) * conv2d.getOutputSize() * conv2d.getOutputSize() * numFilters,
cudaMemcpyDeviceToHost
);
@@ -253,16 +253,16 @@ TEST_F(Conv2dTest, StridedPaddedConvolution) {
activation, input, kernels.data(), d_input
);
EXPECT_EQ(inputSize, conv2d.outputSize);
EXPECT_EQ(inputSize, conv2d.getOutputSize());
d_output = conv2d.forward(d_input);
std::vector<float> output(
conv2d.outputSize * conv2d.outputSize * numFilters
conv2d.getOutputSize() * conv2d.getOutputSize() * numFilters
);
cudaMemcpy(
output.data(), d_output,
sizeof(float) * conv2d.outputSize * conv2d.outputSize * numFilters,
sizeof(float) * conv2d.getOutputSize() * conv2d.getOutputSize() * numFilters,
cudaMemcpyDeviceToHost
);