add assert to check div by zero and rewrite checkKernel function

Samsung · Feb 28, 2024 · 42b3829 · 42b3829
1 parent 41247b0
commit 42b3829
Show file tree

Hide file tree

Showing 10 changed files with 14 additions and 14 deletions.
diff --git a/onert-micro/onert-micro/src/execute/kernels/tests/Conv2D.test.cpp b/onert-micro/onert-micro/src/execute/kernels/tests/Conv2D.test.cpp
@@ -36,7 +36,7 @@ TEST_F(Conv2DTest, Float_P)
 {
   onert_micro::test_model::TestDataFloatConv2D test_data_kernel;
   std::vector<float> output_data_vector =
-    onert_micro::execute::testing::checkSISOKernel<float>(&test_data_kernel);
+    onert_micro::execute::testing::checkKernel<float>(1, &test_data_kernel);
   EXPECT_THAT(output_data_vector,
               FloatArrayNear(test_data_kernel.get_output_data_by_index(0), 0.0001f));
 }

diff --git a/onert-micro/onert-micro/src/execute/kernels/tests/FullyConnected.test.cpp b/onert-micro/onert-micro/src/execute/kernels/tests/FullyConnected.test.cpp
@@ -36,7 +36,7 @@ TEST_F(FullyConnectedTest, Float_P)
 {
   onert_micro::test_model::TestDataFloatFullyConnected test_data_kernel;
   std::vector<float> output_data_vector =
-    onert_micro::execute::testing::checkSISOKernel<float>(&test_data_kernel);
+    onert_micro::execute::testing::checkKernel<float>(1, &test_data_kernel);
   EXPECT_THAT(output_data_vector, test_data_kernel.get_output_data_by_index(0));
 }
 

diff --git a/onert-micro/onert-micro/src/execute/kernels/tests/Gather.test.cpp b/onert-micro/onert-micro/src/execute/kernels/tests/Gather.test.cpp
@@ -37,15 +37,15 @@ TEST_F(GatherTest, Gather_Float_P)
 {
   onert_micro::test_model::TestDataFloatGather test_data_kernel;
   std::vector<float> output_data_vector =
-    onert_micro::execute::testing::checkSISOKernel<float>(&test_data_kernel);
+    onert_micro::execute::testing::checkKernel<float>(1, &test_data_kernel);
   EXPECT_THAT(output_data_vector, test_data_kernel.get_output_data_by_index(0));
 }
 
 TEST_F(GatherTest, Gather_Int_P)
 {
   onert_micro::test_model::TestDataIntGather test_data_kernel;
   std::vector<int32_t> output_data_vector =
-    onert_micro::execute::testing::checkSISOKernel<int32_t>(&test_data_kernel);
+    onert_micro::execute::testing::checkKernel<int32_t>(1, &test_data_kernel);
   EXPECT_THAT(output_data_vector, test_data_kernel.get_output_data_by_index(0));
 }
 

diff --git a/onert-micro/onert-micro/src/execute/kernels/tests/MaxPool2D.test.cpp b/onert-micro/onert-micro/src/execute/kernels/tests/MaxPool2D.test.cpp
@@ -36,7 +36,7 @@ TEST_F(MaxPool2DTest, Float_P)
 {
   onert_micro::test_model::TestDataFloatMaxPool2D test_data_kernel;
   std::vector<float> output_data_vector =
-    onert_micro::execute::testing::checkSISOKernel<float>(&test_data_kernel);
+    onert_micro::execute::testing::checkKernel<float>(1, &test_data_kernel);
   EXPECT_THAT(output_data_vector, test_data_kernel.get_output_data_by_index(0));
 }
 

diff --git a/onert-micro/onert-micro/src/execute/kernels/tests/Pack.test.cpp b/onert-micro/onert-micro/src/execute/kernels/tests/Pack.test.cpp
@@ -35,15 +35,15 @@ TEST_F(PackTest, Float_P)
 {
   test_model::TestDataFloatPack test_data_kernel;
   std::vector<float> output_data_vector =
-    onert_micro::execute::testing::checkTISOKernel<float>(&test_data_kernel);
+    onert_micro::execute::testing::checkKernel<float>(2, &test_data_kernel);
   EXPECT_THAT(output_data_vector, test_data_kernel.get_output_data_by_index(0));
 }
 
 TEST_F(PackTest, Int_P)
 {
   test_model::TestDataIntPack test_data_kernel;
   std::vector<int32_t> output_data_vector =
-    onert_micro::execute::testing::checkTISOKernel<int32_t>(&test_data_kernel);
+    onert_micro::execute::testing::checkKernel<int32_t>(2, &test_data_kernel);
   EXPECT_THAT(output_data_vector, test_data_kernel.get_output_data_by_index(0));
 }
 

diff --git a/onert-micro/onert-micro/src/execute/kernels/tests/ReduceProd.test.cpp b/onert-micro/onert-micro/src/execute/kernels/tests/ReduceProd.test.cpp
@@ -36,15 +36,15 @@ TEST_F(ReduceProdTest, Reduce_Prod_Float_P)
 {
   onert_micro::test_model::TestDataFloatReduceProd test_data_kernel;
   std::vector<float> output_data_vector =
-    onert_micro::execute::testing::checkSISOKernel<float>(&test_data_kernel);
+    onert_micro::execute::testing::checkKernel<float>(1, &test_data_kernel);
   EXPECT_THAT(output_data_vector, test_data_kernel.get_output_data_by_index(0));
 }
 
 TEST_F(ReduceProdTest, Reduce_Prod_Int_P)
 {
   onert_micro::test_model::TestDataIntReduceProd test_data_kernel;
   std::vector<int32_t> output_data_vector =
-    onert_micro::execute::testing::checkSISOKernel<int32_t>(&test_data_kernel);
+    onert_micro::execute::testing::checkKernel<int32_t>(1, &test_data_kernel);
   EXPECT_THAT(output_data_vector, test_data_kernel.get_output_data_by_index(0));
 }
 

diff --git a/onert-micro/onert-micro/src/execute/kernels/tests/Reshape.test.cpp b/onert-micro/onert-micro/src/execute/kernels/tests/Reshape.test.cpp
@@ -35,14 +35,14 @@ TEST_F(ReshapeTest, MainTest_P)
 {
   onert_micro::test_model::TestDataReshapeKernel<float> test_data_kernel(false);
   std::vector<float> output_data_vector =
-    onert_micro::execute::testing::checkSISOKernel<float>(&test_data_kernel);
+    onert_micro::execute::testing::checkKernel<float>(1, &test_data_kernel);
   EXPECT_THAT(output_data_vector, test_data_kernel.get_output_data_by_index(0));
 }
 
 TEST_F(ReshapeTest, MainTest_NEG)
 {
   onert_micro::test_model::TestDataReshapeKernel<float> test_data_kernel(true);
-  EXPECT_DEATH(checkSISOKernel(&test_data_kernel), "");
+  EXPECT_DEATH(checkKernel(1, &test_data_kernel), "");
 }
 
 } // namespace testing

diff --git a/onert-micro/onert-micro/src/execute/kernels/tests/Shape.test.cpp b/onert-micro/onert-micro/src/execute/kernels/tests/Shape.test.cpp
@@ -36,7 +36,7 @@ TEST_F(ShapeTest, MainTest_P)
 {
   onert_micro::test_model::TestDataShapeKernel<float, int32_t> test_data_kernel;
   std::vector<int32_t> output_data_vector =
-    onert_micro::execute::testing::checkSISOKernel<float, int32_t>(&test_data_kernel);
+    onert_micro::execute::testing::checkKernel<float, int32_t>(1, &test_data_kernel);
   EXPECT_THAT(output_data_vector, test_data_kernel.get_output_data_by_index(0));
 }
 

diff --git a/onert-micro/onert-micro/src/execute/kernels/tests/StridedSlice.test.cpp b/onert-micro/onert-micro/src/execute/kernels/tests/StridedSlice.test.cpp
@@ -35,7 +35,7 @@ TEST_F(StridedSliceTest, MainTest_P)
 {
   onert_micro::test_model::TestDataStridedSliceKernel<float> test_data_kernel;
   std::vector<float> output_data_vector =
-    onert_micro::execute::testing::checkSISOKernel<float>(&test_data_kernel);
+    onert_micro::execute::testing::checkKernel<float>(1, &test_data_kernel);
   EXPECT_THAT(output_data_vector, test_data_kernel.get_output_data_by_index(0));
 }
 

diff --git a/onert-micro/onert-micro/src/execute/kernels/tests/UnidirectionalSequenceLSTM.test.cpp b/onert-micro/onert-micro/src/execute/kernels/tests/UnidirectionalSequenceLSTM.test.cpp
@@ -35,7 +35,7 @@ TEST_F(UnidirLSTMTest, Float_P)
 {
   onert_micro::test_model::TestDataFloatUnidirectionalLSTM test_data_kernel;
   std::vector<float> output_data_vector =
-    onert_micro::execute::testing::checkSISOKernel<float>(&test_data_kernel);
+    onert_micro::execute::testing::checkKernel<float>(1, &test_data_kernel);
   EXPECT_THAT(output_data_vector,
               FloatArrayNear(test_data_kernel.get_output_data_by_index(0), 0.0001f));
 }