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 activations enum

Browse Source
This commit is contained in:
Matúš Námešný
2024-03-03 15:24:54 +01:00
parent 7e4460cc5e
commit f37320594a
4 changed files with 33 additions and 22 deletions
Download Patch File Download Diff File Expand all files Collapse all files

13
test/layers/test_dense.cu
View File

@@ -17,7 +17,7 @@ class DenseLayerTest : public CublasTestFixture {
std::vector<float>& biases,
float*& d_input,
float*& d_output,
std::string activation
Activation activation
) {
// Create Dense layer
Layers::Dense denseLayer(
@@ -63,7 +63,7 @@ TEST_F(DenseLayerTest, Init) {
// std::cout << "Dense layer: input size = " << inputSize << ",
// output size = " << outputSize << std::endl;
Layers::Dense denseLayer(
inputSize, outputSize, "sigmoid", cublasHandle
inputSize, outputSize, SIGMOID, cublasHandle
);
}
}
@@ -81,7 +81,7 @@ TEST_F(DenseLayerTest, setWeights) {
{1.3f, 0.5f, 0.0f, 1.7f}
};
Layers::Dense denseLayer(inputSize, outputSize, "sigmoid", cublasHandle);
Layers::Dense denseLayer(inputSize, outputSize, SIGMOID, cublasHandle);
denseLayer.setWeights(weights);
}
@@ -108,8 +108,7 @@ TEST_F(DenseLayerTest, ForwardUnitWeightMatrixLinear) {
float* d_output;
Layers::Dense denseLayer = commonTestSetup(
inputSize, outputSize, input, weights, biases, d_input, d_output,
"linear"
inputSize, outputSize, input, weights, biases, d_input, d_output, LINEAR
);
denseLayer.forward(d_input, d_output);
@@ -145,7 +144,7 @@ TEST_F(DenseLayerTest, ForwardRandomWeightMatrixRelu) {
float* d_output;
Layers::Dense denseLayer = commonTestSetup(
inputSize, outputSize, input, weights, biases, d_input, d_output, "relu"
inputSize, outputSize, input, weights, biases, d_input, d_output, RELU
);
denseLayer.forward(d_input, d_output);
@@ -188,7 +187,7 @@ TEST_F(DenseLayerTest, ForwardRandomWeightMatrixSigmoid) {
Layers::Dense denseLayer = commonTestSetup(
inputSize, outputSize, input, weights, biases, d_input, d_output,
"sigmoid"
SIGMOID
);
denseLayer.forward(d_input, d_output);