Mathis/CUDANet
1
0
Fork 0
mirror of https://github.com/lordmathis/CUDANet.git synced 2025-11-06 01:34:22 +00:00
Code Issues Packages Projects Releases Wiki Activity

Restructure cuda backend

Browse Source
This commit is contained in:
Matúš Námešný
2024-09-05 22:23:47 +02:00
parent 65727dfee8
commit f8220f0ec1
19 changed files with 69 additions and 16 deletions
Download Patch File Download Diff File Expand all files Collapse all files

14
include/layers/add.cuh → include/layers/add.hpp
View File

@@ -25,13 +25,23 @@ class Add {
* @param d_inputB Device pointer to the second input
*
*/
void forward(const float* d_inputA, const float* d_inputB);
float* forward(const float* inputA, const float* inputB);
private:
int inputSize;
float* output;
float* forwardCPU(const float* inputA, const float* inputB);
#ifdef USE_CUDA
float* d_output;
int gridSize;
float* d_output;
float* forwardCUDA(const float* d_inputA, const float* d_inputB);
void initCUDA();
void delCUDA();
#endif
};
} // namespace CUDANet::Layers