Add activations enum

2025-11-05 17:34:21 +00:00 · 2024-03-03 15:24:54 +01:00
parent 7e4460cc5e
commit f37320594a
4 changed files with 33 additions and 22 deletions
--- a/include/kernels/activations.cuh
+++ b/include/kernels/activations.cuh
@@ -10,4 +10,10 @@ relu_kernel(const float* __restrict__ src, float* __restrict__ dst, int len);
 __global__ void
 linear_kernel(const float* __restrict__ src, float* __restrict__ dst, int len);

+enum Activation {
+    SIGMOID,
+    RELU,
+    LINEAR
+};
+
 #endif  // ACTIVATIONS_H