Split tests to cpu and cuda

This commit is contained in:
2024-09-07 21:24:34 +02:00
parent f8220f0ec1
commit 591507ed21
16 changed files with 29 additions and 10 deletions

View File

@@ -0,0 +1,19 @@
#include <cuda_runtime_api.h>
#include <gtest/gtest.h>
#include "input.cuh"
TEST(InputLayerTest, InputForward) {
std::vector<float> input = {0.573f, 0.619f, 0.732f, 0.055f, 0.243f, 0.316f};
CUDANet::Layers::Input inputLayer(6);
float* d_output = inputLayer.forward(input.data());
std::vector<float> output(6);
cudaError_t cudaStatus = cudaMemcpy(
output.data(), d_output, sizeof(float) * 6, cudaMemcpyDeviceToHost
);
EXPECT_EQ(cudaStatus, cudaSuccess);
EXPECT_EQ(input, output);
}