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

Allocate activation on heap

Browse Source
This commit is contained in:
Matúš Námešný
2024-04-22 18:59:16 +02:00
parent 26cea9b12c
commit a32c737785
10 changed files with 17 additions and 15 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
include/layers/avg_pooling.cuh
View File

@@ -38,7 +38,7 @@ class AvgPooling2D : public SequentialLayer {
float* d_output;
Activation activation;
Activation* activation;
};
} // namespace CUDANet::Layers

3
include/layers/conv2d.cuh
View File

@@ -120,8 +120,7 @@ class Conv2d : public WeightedLayer {
float* d_weights;
float* d_biases;
// Kernels
Activation activation;
Activation* activation;
/**
* @brief Initialize weights of the convolutional layer with zeros

2
include/layers/dense.cuh
View File

@@ -77,7 +77,7 @@ class Dense : public WeightedLayer {
std::vector<float> weights;
std::vector<float> biases;
Layers::Activation activation;
Layers::Activation* activation;
// Precompute kernel launch parameters
unsigned int forwardGridSize;

2
include/layers/max_pooling.cuh
View File

@@ -38,7 +38,7 @@ class MaxPooling2D : public SequentialLayer {
float* d_output;
Activation activation;
Activation* activation;
};
} // namespace CUDANet::Layers