From a3fad2d3a96b8cb68cfd38ef8c09d543ae45444a Mon Sep 17 00:00:00 2001 From: Dheeraj Peri Date: Thu, 6 Feb 2025 15:42:58 -0800 Subject: [PATCH] chore: disable instance_norm test --- tests/core/conversion/converters/test_instance_norm.cpp | 3 +++ 1 file changed, 3 insertions(+) diff --git a/tests/core/conversion/converters/test_instance_norm.cpp b/tests/core/conversion/converters/test_instance_norm.cpp index 8f9904ef84..3b3ae130be 100644 --- a/tests/core/conversion/converters/test_instance_norm.cpp +++ b/tests/core/conversion/converters/test_instance_norm.cpp @@ -4,6 +4,7 @@ #include "tests/util/util.h" #include "torch/csrc/jit/ir/irparser.h" +#ifndef DISABLE_TEST_IN_CI // Tensor instance_norm( // const Tensor& input, // const c10::optional& weight_opt /* optional */, @@ -101,3 +102,5 @@ TEST(Converters, ATenInstanceNormRunningStatsConvertsCorrectly) { auto trt_results = torch_tensorrt::tests::util::RunGraphEngine(g, params, {trt_in}); ASSERT_TRUE(torch_tensorrt::tests::util::almostEqual(jit_results[0], trt_results[0].reshape_as(jit_results[0]))); } + +#endif \ No newline at end of file