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

Implement output layer

Browse Source
This commit is contained in:
Matúš Námešný
2024-03-21 23:07:46 +01:00
parent e46d5d3f76
commit 90fb104dae
8 changed files with 92 additions and 9 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
test/layers/test_input.cu
View File

@@ -1,16 +1,17 @@
#include <cuda_runtime_api.h>
#include <gtest/gtest.h>
#include "cuda_helper.cuh"
#include "input.cuh"
TEST(InputLayerTest, Init) {
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);
CUDA_CHECK(cudaMemcpy(
cudaError_t cudaStatus = cudaMemcpy(
output.data(), d_output, sizeof(float) * 6, cudaMemcpyDeviceToHost
));
);
EXPECT_EQ(cudaStatus, cudaSuccess);
EXPECT_EQ(input, output);
}

24
test/layers/test_output.cu Normal file
View File

@@ -0,0 +1,24 @@
#include <cuda_runtime_api.h>
#include <gtest/gtest.h>
#include "output.cuh"
TEST(OutputLayerTest, OutputForward) {
cudaError_t cudaStatus;
std::vector<float> input = {0.573f, 0.619f, 0.732f, 0.055f, 0.243f, 0.316f};
float* d_input;
cudaStatus = cudaMalloc((void**)&d_input, sizeof(float) * 6);
EXPECT_EQ(cudaStatus, cudaSuccess);
cudaStatus = cudaMemcpy(
d_input, input.data(), sizeof(float) * 6, cudaMemcpyHostToDevice
);
EXPECT_EQ(cudaStatus, cudaSuccess);
CUDANet::Layers::Output outputLayer(6);
float* h_output = outputLayer.forward(d_input);
for (int i = 0; i < 6; ++i) {
EXPECT_EQ(input[i], h_output[i]);
}
}