test: properly disabling parameterized tests

Signed-off-by: Naren Dasan <naren@narendasan.com> Signed-off-by: Naren Dasan <narens@nvidia.com>
pytorch · Jul 6, 2023 · f14bc21 · f14bc21
1 parent 4f2c708
commit f14bc21
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/tests/cpp/test_modules_as_engines.cpp b/tests/cpp/test_modules_as_engines.cpp
@@ -17,7 +17,11 @@ TEST_P(CppAPITests, ModuleAsEngineIsClose) {
   ASSERT_TRUE(torch_tensorrt::tests::util::cosineSimEqual(jit_results[0], trt_results[0].reshape_as(jit_results[0])));
 }
 
-#ifndef DISABLE_TEST_IN_CI
+#ifdef DISABLE_TEST_IN_CI
+
+GTEST_ALLOW_UNINSTANTIATED_PARAMETERIZED_TEST(CppAPITests);
+
+#else
 
 INSTANTIATE_TEST_SUITE_P(
     ModuleAsEngineForwardIsCloseSuite,