tests: properly disable parameterized tests in CI

narendasan · narendasan · commit b45379958f2e · 2023-07-25T16:51:38.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;

Update config.yml

Fix issue with CMake build
diff --git a/.circleci/config.yml b/.circleci/config.yml
@@ -1254,7 +1254,7 @@ jobs:
           platform: "x86_64"
           cudnn-version: << parameters.cudnn-version >>
           trt-version-short:  << parameters.trt-version-short >>
-          bazel-platform: "x86_64"
+          bazel-platform: "amd64"
       - install-cmake:
           version: 3.24.1
       - install-torch-from-index:
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,`