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

Initial activations implementation

Browse Source
This commit is contained in:
Matúš Námešný
2024-02-27 00:24:57 +01:00
parent 6e99525ad0
commit 5e1e0ed1d1
9 changed files with 104 additions and 24 deletions
Download Patch File Download Diff File Expand all files Collapse all files

14
include/functions/activations.cuh Normal file
View File

@@ -0,0 +1,14 @@
#include <functional>
#ifndef ACTIVATIONS_H
#define ACTIVATIONS_H
__device__ float sigmoid(float a);
__device__ float relu(float a);
__device__ float linear(float a);
__global__ void sigmoid_kernel(const float* __restrict__ src, float* __restrict__ dst, int len);
__global__ void relu_kernel(const float* __restrict__ src, float* __restrict__ dst, int len);
__global__ void linear_kernel(const float* __restrict__ src, float* __restrict__ dst, int len);
#endif // ACTIVATIONS_H