Move cuda source to separate directory

This commit is contained in:
2024-09-05 17:39:09 +02:00
parent 9973ebc477
commit 65727dfee8
16 changed files with 0 additions and 0 deletions

31
src/cuda/layers/input.cu Normal file
View File

@@ -0,0 +1,31 @@
#include "cuda_helper.cuh"
#include "input.cuh"
using namespace CUDANet::Layers;
Input::Input(int inputSize) : inputSize(inputSize) {
d_output = nullptr;
CUDA_CHECK(cudaMalloc((void**)&d_output, sizeof(float) * inputSize));
}
Input::~Input() {
cudaFree(d_output);
}
float* Input::forward(const float* input) {
CUDA_CHECK(cudaMemcpy(
d_output, input, sizeof(float) * inputSize, cudaMemcpyHostToDevice
));
CUDA_CHECK(cudaDeviceSynchronize());
return d_output;
}
int Input::getOutputSize() {
return inputSize;
}
int Input::getInputSize() {
return inputSize;
}