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

Initial activations implementation

Browse Source
This commit is contained in:
Matúš Námešný
2024-02-27 00:24:57 +01:00
parent 6e99525ad0
commit 5e1e0ed1d1
9 changed files with 104 additions and 24 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
test/layers/test_dense.cu
View File

@@ -2,6 +2,7 @@
#include <cuda_runtime_api.h>
#include <driver_types.h>
#include <iostream>
#include "activations.cuh"
#include "dense.cuh"
#include "test_cublas_fixture.cuh"
@@ -9,7 +10,7 @@ class DenseLayerTest : public CublasTestFixture {
protected:
Layers::Dense commonTestSetup(int inputSize, int outputSize, std::vector<float>& input, std::vector<std::vector<float>>& weights, std::vector<float>& biases, float*& d_input, float*& d_output) {
// Create Dense layer
Layers::Dense denseLayer(inputSize, outputSize, cublasHandle);
Layers::Dense denseLayer(inputSize, outputSize, "linear", cublasHandle);
// Set weights and biases
denseLayer.setWeights(weights);
@@ -48,7 +49,7 @@ TEST_F(DenseLayerTest, Init) {
int outputSize = j;
// std::cout << "Dense layer: input size = " << inputSize << ", output size = " << outputSize << std::endl;
Layers::Dense denseLayer(inputSize, outputSize, cublasHandle);
Layers::Dense denseLayer(inputSize, outputSize, "linear", cublasHandle);
}
}
}
@@ -67,7 +68,7 @@ TEST_F(DenseLayerTest, setWeights) {
{1.3f, 0.5f, 0.0f, 1.7f}
};
Layers::Dense denseLayer(inputSize, outputSize, cublasHandle);
Layers::Dense denseLayer(inputSize, outputSize, "linear", cublasHandle);
denseLayer.setWeights(weights);