Migrate Activation layer

This commit is contained in:
2024-09-08 12:49:13 +02:00
parent 591507ed21
commit 0dca8348bd
4 changed files with 46 additions and 8 deletions

View File

@@ -47,11 +47,21 @@ class Activation {
private:
ActivationType activationType;
int length;
#ifdef USE_CUDA
int gridSize;
float* d_softmax_sum;
float* d_max;
void activateCUDA(float* d_input);
void initCUDA();
void delCUDA();
#else
void activateCPU(float* input);
#endif
};