explicitly use floats, get rid of compiler warnings

fm94 · Jun 2, 2024 · b5fb456 · b5fb456
1 parent dbfcc01
commit b5fb456
Show file tree

Hide file tree

Showing 3 changed files with 15 additions and 15 deletions.
diff --git a/tests/test_dense.cpp b/tests/test_dense.cpp
@@ -8,10 +8,10 @@ using namespace Layer;
 TEST(DenseLayerTest, ForwardPass)
 {
     Dense dense(3, 2);
-    dense.set_learning_rate(0.01);
+    dense.set_learning_rate(0.01f);
 
     Eigen::MatrixXf input(1, 3);
-    input << 1.0, 2.0, 3.0;
+    input << 1.0f, 2.0f, 3.0f;
     Eigen::MatrixXf output(1, 2);
 
     dense.forward(input, output);
@@ -24,16 +24,16 @@ TEST(DenseLayerTest, ForwardPass)
 TEST(DenseLayerTest, BackwardPass)
 {
     Dense dense(3, 2);
-    dense.set_learning_rate(0.01);
+    dense.set_learning_rate(0.01f);
 
     Eigen::MatrixXf input(1, 3);
-    input << 1.0, 2.0, 3.0;
+    input << 1.0f, 2.0f, 3.0f;
     Eigen::MatrixXf output(1, 2);
 
     dense.forward(input, output);
 
     Eigen::MatrixXf dout(1, 2);
-    dout << 0.1, 0.2;
+    dout << 0.1f, 0.2f;
     Eigen::MatrixXf din(1, 3);
     dense.backward(dout, din);
 

diff --git a/tests/test_relu.cpp b/tests/test_relu.cpp
@@ -9,31 +9,31 @@ TEST(ReLULayerTest, ForwardPass)
 {
     ReLU            relu;
     Eigen::MatrixXf input(1, 3);
-    input << 1.0, -2.0, 3.0;
+    input << 1.0f, -2.0f, 3.0f;
     Eigen::MatrixXf output(1, 3);
 
     relu.forward(input, output);
 
     Eigen::MatrixXf expected_output(1, 3);
-    expected_output << 1.0, 0.0, 3.0;
+    expected_output << 1.0f, 0.0f, 3.0f;
     expectEigenMatrixNear(output, expected_output);
 }
 
 TEST(ReLULayerTest, BackwardPass)
 {
     ReLU            relu;
     Eigen::MatrixXf input(1, 3);
-    input << 1.0, -2.0, 3.0;
+    input << 1.0f, -2.0f, 3.0f;
     Eigen::MatrixXf output(1, 3);
 
     relu.forward(input, output);
 
     Eigen::MatrixXf dout(1, 3);
-    dout << 0.1, 0.2, 0.3;
+    dout << 0.1f, 0.2f, 0.3f;
     Eigen::MatrixXf din(1, 3);
     relu.backward(dout, din);
 
     Eigen::MatrixXf expected_din(1, 3);
-    expected_din << 0.1, 0.0, 0.3;
+    expected_din << 0.1f, 0.0f, 0.3f;
     expectEigenMatrixNear(din, expected_din);
 }
diff --git a/tests/test_softmax.cpp b/tests/test_softmax.cpp
@@ -9,13 +9,13 @@ TEST(SoftmaxLayerTest, ForwardPass)
 {
     Softmax         softmax;
     Eigen::MatrixXf input(2, 2);
-    input << 1.0, 2.0, 3.0, 4.0;
+    input << 1.0f, 2.0f, 3.0f, 4.0f;
 
     Eigen::MatrixXf output;
     softmax.forward(input, output);
 
     Eigen::MatrixXf expected_output(2, 2);
-    expected_output << 0.268941, 0.731059, 0.268941, 0.731059;
+    expected_output << 0.268941f, 0.731059f, 0.268941f, 0.731059f;
 
     expectEigenMatrixNear(output, expected_output, 1e-5);
 }
@@ -24,20 +24,20 @@ TEST(SoftmaxLayerTest, BackwardPass)
 {
     Softmax         softmax;
     Eigen::MatrixXf input(2, 2);
-    input << 1.0, 2.0, 3.0, 4.0;
+    input << 1.0f, 2.0f, 3.0f, 4.0f;
 
     Eigen::MatrixXf output;
     softmax.forward(input, output);
 
     Eigen::MatrixXf out_grad(2, 2);
-    out_grad << 0.1, -0.1, 0.1, -0.1;
+    out_grad << 0.1f, -0.1f, 0.1f, -0.1f;
 
     Eigen::MatrixXf in_grad;
     softmax.backward(out_grad, in_grad);
 
     // manually calculated
     Eigen::MatrixXf expected_in_grad(2, 2);
-    expected_in_grad << 0.0393, -0.0393, 0.0393, -0.0393;
+    expected_in_grad << 0.0393f, -0.0393f, 0.0393f, -0.0393f;
 
     expectEigenMatrixNear(in_grad, expected_in_grad, 1e-3);
 }