tests: properly disable parameterized tests in CI

narendasan · narendasan · commit 05815e6418e5 · 2023-07-06T08:33:11.000-07:00
Signed-off-by: Naren Dasan &lt;naren@narendasan.com&gt;
Signed-off-by: Naren Dasan &lt;narens@nvidia.com&gt;

test: properly disabling parameterized tests

Signed-off-by: Naren Dasan &lt;naren@narendasan.com&gt;
Signed-off-by: Naren Dasan &lt;narens@nvidia.com&gt;
diff --git a/tests/cpp/test_compiled_modules.cpp b/tests/cpp/test_compiled_modules.cpp
@@ -45,7 +45,11 @@ TEST_P(CppAPITests, CompiledModuleIsClose) {
   }
 }
 
-#ifndef DISABLE_TEST_IN_CI
+#ifdef DISABLE_TEST_IN_CI
+
+GTEST_ALLOW_UNINSTANTIATED_PARAMETERIZED_TEST(CppAPITests);
+
+#else
 
 INSTANTIATE_TEST_SUITE_P(
     CompiledModuleForwardIsCloseSuite,
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,

Original file line number	Diff line number	Diff line change
`@@ -45,7 +45,11 @@ TEST_P(CppAPITests, CompiledModuleIsClose) {`
`45`	`45`	`}`
`46`	`46`	`}`
`47`	`47`
`48`		`-#ifndef DISABLE_TEST_IN_CI`
	`48`	`+#ifdef DISABLE_TEST_IN_CI`
	`49`	`+`
	`50`	`+GTEST_ALLOW_UNINSTANTIATED_PARAMETERIZED_TEST(CppAPITests);`
	`51`	`+`
	`52`	`+#else`
`49`	`53`
`50`	`54`	`INSTANTIATE_TEST_SUITE_P(`
`51`	`55`	`CompiledModuleForwardIsCloseSuite,`