Start conv test implementation

This commit is contained in:
2024-03-07 22:03:05 +01:00
parent 7e75943a6b
commit 69ccba2dad
3 changed files with 48 additions and 12 deletions

View File

@@ -0,0 +1,35 @@
#include <cuda_runtime_api.h>
#include <gtest/gtest.h>
#include <iostream>
#include "conv2d.cuh"
TEST(Conv2dTest, ValidPadding) {
int inputSize = 3;
int inputChannels = 1;
int kernelSize = 3;
int stride = 1;
std::string padding = "VALID";
int numFilters = 1;
Activation activation = LINEAR;
Layers::Conv2d conv2d(
inputSize,
inputChannels,
kernelSize,
stride,
padding,
numFilters,
activation
);
int outputSize = (inputSize - kernelSize) / stride + 1;
EXPECT_EQ(outputSize, conv2d.outputSize);
std::vector<float> input(inputSize * inputSize * inputChannels);
std::vector<float> output(outputSize * outputSize * numFilters);
std::vector<float> kernels(kernelSize * kernelSize * inputChannels * numFilters);
}