Added the tests for all 4 Binary Operators

Author: Neel-Shah-29

tmva/sofie/test/TestCustomModelsFromONNX.cxx

@@ -12,6 +12,18 @@
 #include "LinearWithSelu_FromONNX.hxx"
 #include "input_models/references/LinearWithSelu.ref.hxx"
 
+#include "Sub_FromONNX.hxx"
+#include "input_models/references/Sub.ref.hxx"
+
+#include "Add_FromONNX.hxx"
+#include "input_models/references/Add.ref.hxx"
+
+#include "Mul_FromONNX.hxx"
+#include "input_models/references/Mul.ref.hxx"
+
+#include "Div_FromONNX.hxx"
+#include "input_models/references/Div.ref.hxx"
+
 #include "LinearWithLeakyRelu_FromONNX.hxx"
 #include "input_models/references/LinearWithLeakyRelu.ref.hxx"
 
@@ -134,6 +146,110 @@ TEST(ONNX, Linear32)
    }
 }
 
+TEST(ONNX, Sub)
+   {
+      constexpr float TOLERANCE = DEFAULT_TOLERANCE;
+
+      // Preparing the standard input
+      std::vector<float> input1({
+         1, 2
+      });
+      std::vector<float> input2({
+         0, 1
+      });
+      TMVA_SOFIE_Sub::Session s("Sub_FromONNX.dat");
+
+      std::vector<float> output = s.infer(input2.data(),input1.data());
+
+      // Checking output size
+      EXPECT_EQ(output.size(), sizeof(Sub_ExpectedOutput::outputs) / sizeof(float));
+
+      float *correct = Sub_ExpectedOutput::outputs;
+
+      // Checking every output value, one by one
+      for (size_t i = 0; i < output.size(); ++i) {
+         EXPECT_LE(std::abs(output[i] - correct[i]), TOLERANCE);
+      }
+   }
+
+
+TEST(ONNX, Add)
+   {
+      constexpr float TOLERANCE = DEFAULT_TOLERANCE;
+
+      // Preparing the standard input
+      std::vector<float> input1({
+         1, 2
+      });
+      std::vector<float> input2({
+         0, 1
+      });
+      TMVA_SOFIE_Add::Session s("Add_FromONNX.dat");
+
+      std::vector<float> output = s.infer(input1.data(),input2.data());
+
+      // Checking output size
+      EXPECT_EQ(output.size(), sizeof(Add_ExpectedOutput::outputs) / sizeof(float));
+
+      float *correct = Add_ExpectedOutput::outputs;
+
+      // Checking every output value, one by one
+      for (size_t i = 0; i < output.size(); ++i) {
+         EXPECT_LE(std::abs(output[i] - correct[i]), TOLERANCE);
+      }
+   }
+
+TEST(ONNX, Mul)
+   {
+      constexpr float TOLERANCE = DEFAULT_TOLERANCE;
+
+      // Preparing the standard input
+      std::vector<float> input1({
+         1, 2
+      });
+      std::vector<float> input2({
+         0, 1
+      });
+      TMVA_SOFIE_Mul::Session s("Mul_FromONNX.dat");
+
+      std::vector<float> output = s.infer(input1.data(),input2.data());
+
+      // Checking output size
+      EXPECT_EQ(output.size(), sizeof(Mul_ExpectedOutput::outputs) / sizeof(float));
+
+      float *correct = Mul_ExpectedOutput::outputs;
+
+      // Checking every output value, one by one
+      for (size_t i = 0; i < output.size(); ++i) {
+         EXPECT_LE(std::abs(output[i] - correct[i]), TOLERANCE);
+      }
+   }
+
+TEST(ONNX, Div)
+   {
+      constexpr float TOLERANCE = DEFAULT_TOLERANCE;
+
+      // Preparing the standard input
+      std::vector<float> input1({
+         4, 2
+      });
+      std::vector<float> input2({
+         2, 2
+      });
+      TMVA_SOFIE_Div::Session s("Div_FromONNX.dat");
+
+      std::vector<float> output = s.infer(input2.data(),input1.data());
+
+      // Checking output size
+      EXPECT_EQ(output.size(), sizeof(Div_ExpectedOutput::outputs) / sizeof(float));
+
+      float *correct = Div_ExpectedOutput::outputs;
+
+      // Checking every output value, one by one
+      for (size_t i = 0; i < output.size(); ++i) {
+         EXPECT_LE(std::abs(output[i] - correct[i]), TOLERANCE);
+      }
+   }
 
 TEST(ONNX, Linear64)
 {

tmva/sofie/test/input_models/Add.onnx

@@ -0,0 +1,16 @@
+pytorch1.11.0:�

+)
+onnx::Add_0
+onnx::Add_1
2Add_0"Addtorch-jit-exportZ
+onnx::Add_0
+
+

+Z
+onnx::Add_1
+
+

+b
+
2
+
+

+B	

tmva/sofie/test/input_models/Div.onnx

@@ -0,0 +1,16 @@
+pytorch1.11.0:�

+)
+onnx::Div_0
+onnx::Div_1
2Div_0"Divtorch-jit-exportZ
+onnx::Div_0
+
+

+Z
+onnx::Div_1
+
+

+b
+
2
+
+

+B	

tmva/sofie/test/input_models/Mul.onnx

@@ -0,0 +1,16 @@
+pytorch1.11.0:�

+)
+onnx::Mul_0
+onnx::Mul_1
2Mul_0"Multorch-jit-exportZ
+onnx::Mul_0
+
+

+Z
+onnx::Mul_1
+
+

+b
+
2
+
+

+B	

tmva/sofie/test/input_models/Sub.onnx

@@ -0,0 +1,16 @@
+pytorch1.11.0:�

+)
+onnx::Sub_0
+onnx::Sub_1
2Sub_0"Subtorch-jit-exportZ
+onnx::Sub_0
+
+

+Z
+onnx::Sub_1
+
+

+b
+
2
+
+

+B	

tmva/sofie/test/input_models/references/Add.ref.hxx

@@ -0,0 +1,5 @@
+namespace Add_ExpectedOutput{
+	float outputs[] = {
+        1, 3
+	};
+} // namespace Add_ExpectedOutput

tmva/sofie/test/input_models/references/Div.ref.hxx

@@ -0,0 +1,5 @@
+namespace Div_ExpectedOutput{
+	float outputs[] = {
+        2, 1
+	};
+} // namespace Div_ExpectedOutput

tmva/sofie/test/input_models/references/Mul.ref.hxx

@@ -0,0 +1,5 @@
+namespace Mul_ExpectedOutput{
+	float outputs[] = {
+        0, 2
+	};
+} // namespace Mul_ExpectedOutput

tmva/sofie/test/input_models/references/Sub.ref.hxx

@@ -0,0 +1,5 @@
+namespace Sub_ExpectedOutput{
+	float outputs[] = {
+        1, 1
+	};
+} // namespace Sub_ExpectedOutput