Mathis/CUDANet
1
0
Fork 0
mirror of https://github.com/lordmathis/CUDANet.git synced 2025-11-06 01:34:22 +00:00
Code Issues Packages Projects Releases Wiki Activity

Format code

Browse Source
This commit is contained in:
Matúš Námešný
2024-02-27 21:49:05 +01:00
parent 19ee20ea66
commit e267f08a2f
2 changed files with 3 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
test/kernels/test_activations.cu
View File

@@ -14,7 +14,6 @@ class ActivationsTest : public CublasTestFixture {
}; };
TEST_F(ActivationsTest, SigmoidSanityCheck) { TEST_F(ActivationsTest, SigmoidSanityCheck) {
float input[3] = {-100.0f, 0.0f, 100.0f}; float input[3] = {-100.0f, 0.0f, 100.0f};
std::vector<float> expected_output = {0.0f, 0.5f, 1.0f}; std::vector<float> expected_output = {0.0f, 0.5f, 1.0f};
@@ -37,7 +36,8 @@ TEST_F(ActivationsTest, SigmoidSanityCheck) {
std::vector<float> output(3); std::vector<float> output(3);
cublasStatus = cublasGetVector(3, sizeof(float), d_output, 1, output.data(), 1); cublasStatus =
cublasGetVector(3, sizeof(float), d_output, 1, output.data(), 1);
EXPECT_EQ(cublasStatus, CUBLAS_STATUS_SUCCESS); EXPECT_EQ(cublasStatus, CUBLAS_STATUS_SUCCESS);
for (int i = 0; i < 3; i++) { for (int i = 0; i < 3; i++) {

4
test/layers/test_dense.cu
View File

@@ -209,9 +209,7 @@ TEST_F(DenseLayerTest, ForwardRandomWeightMatrixSigmoid) {
}; };
for (int i = 0; i < outputSize; ++i) { for (int i = 0; i < outputSize; ++i) {
EXPECT_NEAR( EXPECT_NEAR(output[i], expectedOutput[i], 1e-5);
output[i], expectedOutput[i], 1e-5
);
} }
commonTestTeardown(d_input, d_output); commonTestTeardown(d_input, d_output);