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

19
test/layers/test_dense.cu
View File

@@ -16,7 +16,7 @@ class DenseLayerTest : public ::testing::Test {
float* biases,
float*& d_input,
float*& d_output,
Activation activation
Layers::Activation activation
) {
// Create Dense layer
Layers::Dense denseLayer(inputSize, outputSize, activation);
@@ -57,7 +57,9 @@ TEST_F(DenseLayerTest, Init) {
int inputSize = i;
int outputSize = j;
Layers::Dense denseLayer(inputSize, outputSize, SIGMOID);
Layers::Dense denseLayer(
inputSize, outputSize, Layers::Activation::SIGMOID
);
}
}
}
@@ -76,7 +78,9 @@ TEST_F(DenseLayerTest, setWeights) {
};
// clang-format on
Layers::Dense denseLayer(inputSize, outputSize, SIGMOID);
Layers::Dense denseLayer(
inputSize, outputSize, Layers::Activation::SIGMOID
);
denseLayer.setWeights(weights.data());
}
@@ -102,7 +106,7 @@ TEST_F(DenseLayerTest, ForwardUnitWeightMatrixLinear) {
Layers::Dense denseLayer = commonTestSetup(
inputSize, outputSize, input, weights.data(), biases.data(), d_input,
d_output, LINEAR
d_output, Layers::Activation::NONE
);
denseLayer.forward(d_input, d_output);
@@ -142,7 +146,8 @@ TEST_F(DenseLayerTest, ForwardRandomWeightMatrixRelu) {
float* d_output;
Layers::Dense denseLayer = commonTestSetup(
inputSize, outputSize, input, weights.data(), biases.data(), d_input, d_output, RELU
inputSize, outputSize, input, weights.data(), biases.data(), d_input,
d_output, Layers::Activation::RELU
);
denseLayer.forward(d_input, d_output);
@@ -186,8 +191,8 @@ TEST_F(DenseLayerTest, ForwardRandomWeightMatrixSigmoid) {
float* d_output;
Layers::Dense denseLayer = commonTestSetup(
inputSize, outputSize, input, weights.data(), biases.data(), d_input, d_output,
SIGMOID
inputSize, outputSize, input, weights.data(), biases.data(), d_input,
d_output, Layers::Activation::SIGMOID
);
denseLayer.forward(d_input, d_output);