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

Add activations test

Browse Source
This commit is contained in:
Matúš Námešný
2024-02-27 20:19:17 +01:00
parent 48ba09b28d
commit b1eb8b5806
6 changed files with 54 additions and 21 deletions
Download Patch File Download Diff File Expand all files Collapse all files

17
include/kernels/activations.cuh Normal file
View File

@@ -0,0 +1,17 @@
#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