From f14bc21394442ce263532cefae1ebd57a9cf0034 Mon Sep 17 00:00:00 2001 From: Naren Dasan <1790613+narendasan@users.noreply.github.com> Date: Mon, 3 Jul 2023 17:14:16 +0000 Subject: [PATCH] test: properly disabling parameterized tests Signed-off-by: Naren Dasan Signed-off-by: Naren Dasan --- tests/cpp/test_modules_as_engines.cpp | 6 +++++- 1 file changed, 5 insertions(+), 1 deletion(-) diff --git a/tests/cpp/test_modules_as_engines.cpp b/tests/cpp/test_modules_as_engines.cpp index b76bae7333..b2832298a3 100644 --- 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,