Revert to not compiling Conv2D model test when AVX enabled

jatinchowdhury18 · Nov 27, 2023 · 1fb6bf1 · 1fb6bf1
1 parent 0334d72
commit 1fb6bf1
Showing 1 changed file with 4 additions and 5 deletions.
diff --git a/tests/functional/conv2d_model_test.cpp b/tests/functional/conv2d_model_test.cpp
@@ -101,8 +101,6 @@ auto loadTemplatedModel()
     return modelT;
 }
 
-
-
 template <typename ModelType>
 void testModelOutputMatchesPythonImplementation(
     ModelType& model,
@@ -153,22 +151,22 @@ TEST(TestConv2D, nonTemplatedModelOutputMatchesPythonImplementation)
 {
 #if RTNEURAL_AVX_ENABLED
     GTEST_SKIP() << "SKIPPING CONV2D MODEL TEST w/ AVX ENABLED...";
-#endif
-
+#else
     auto non_templated_model = loadNonTemplatedModel();
     ASSERT_THAT(non_templated_model, testing::Ne(nullptr));
     testModelOutputMatchesPythonImplementation(
         *non_templated_model,
         computeReceptiveField(*non_templated_model),
         computeTotalPaddedLeftFramesTensorflow(*non_templated_model),
         non_templated_model->getOutSize());
+#endif
 }
 
 TEST(TestConv2D, templatedModelOutputMatchesPythonImplementation)
 {
 #if RTNEURAL_AVX_ENABLED
     GTEST_SKIP() << "SKIPPING CONV2D MODEL TEST w/ AVX ENABLED...";
-#endif
+#else
     auto non_templated_model = loadNonTemplatedModel();
     ASSERT_THAT(non_templated_model, testing::Ne(nullptr));
 
@@ -178,4 +176,5 @@ TEST(TestConv2D, templatedModelOutputMatchesPythonImplementation)
         computeReceptiveField(*non_templated_model),
         computeTotalPaddedLeftFramesTensorflow(*non_templated_model),
         non_templated_model->getOutSize());
+#endif
 }