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

Add Kernels namespace

Browse Source
This commit is contained in:
Matúš Námešný
2024-03-11 21:04:23 +01:00
parent e0178e2d5c
commit d2ab78fbc7
18 changed files with 188 additions and 186 deletions
Download Patch File Download Diff File Expand all files Collapse all files

13
include/kernels/activations.cuh
View File

@@ -1,19 +1,14 @@
#ifndef ACTIVATIONS_H
#define ACTIVATIONS_H
__global__ void
sigmoid_kernel(const float* __restrict__ src, float* __restrict__ dst, int len);
namespace Kernels {
__global__ void
relu_kernel(const float* __restrict__ src, float* __restrict__ dst, int len);
sigmoid(const float* __restrict__ src, float* __restrict__ dst, int len);
__global__ void
linear_kernel(const float* __restrict__ src, float* __restrict__ dst, int len);
relu(const float* __restrict__ src, float* __restrict__ dst, int len);
enum Activation {
SIGMOID,
RELU,
LINEAR
};
} // namespace Kernels
#endif // ACTIVATIONS_H