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

Refactor Backend and Layer interfaces

Browse Source
This commit is contained in:
Matúš Námešný
2025-11-18 18:27:57 +01:00
parent 25670f90c4
commit 6340b27055
23 changed files with 154 additions and 201 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
test/cuda/layers/test_avg_pooling.cu
View File

@@ -54,7 +54,7 @@ class AvgPoolingLayerTest : public ::testing::Test {
d_output = avgPoolingLayer->forward(d_input);
int outputSize = avgPoolingLayer->getOutputSize();
int outputSize = avgPoolingLayer->get_output_size();
std::vector<float> output(outputSize);
cudaStatus = cudaMemcpy(
@@ -229,7 +229,7 @@ class AdaptiveAvgPoolingLayerTest : public ::testing::Test {
d_output = adaptiveAvgPoolingLayer->forward(d_input);
int outputSize = adaptiveAvgPoolingLayer->getOutputSize();
int outputSize = adaptiveAvgPoolingLayer->get_output_size();
std::vector<float> output(outputSize);
cudaStatus = cudaMemcpy(

2
test/cuda/layers/test_max_pooling.cu
View File

@@ -52,7 +52,7 @@ class MaxPoolingLayerTest : public ::testing::Test {
d_output = maxPoolingLayer->forward(d_input);
int outputSize = maxPoolingLayer->getOutputSize();
int outputSize = maxPoolingLayer->get_output_size();
std::vector<float> output(outputSize);
cudaStatus = cudaMemcpy(