From 3b63d85c253c50099c70ba0db6c141b842bc7cda Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Xavier=20Dupr=C3=A9?= Date: Mon, 6 Nov 2023 19:32:26 +0100 Subject: [PATCH] Fix unit test when TVM EP is enabled (#18189) ### Description TestInlinedLocalFunctionNotRemoved checks that local functions are not removed but TVM EP optimizes the whole graph after it is inlined. --- onnxruntime/test/framework/function_test.cc | 6 ++++++ 1 file changed, 6 insertions(+) diff --git a/onnxruntime/test/framework/function_test.cc b/onnxruntime/test/framework/function_test.cc index 41274ee0de..f9a5c76186 100644 --- a/onnxruntime/test/framework/function_test.cc +++ b/onnxruntime/test/framework/function_test.cc @@ -582,7 +582,13 @@ TEST(FunctionTest, TestInlinedLocalFunctionNotRemoved) { // myfun is not removed because it was claimed by InternalTestingEP model_proto = session_object.GetModel().ToProto(); +#ifdef USE_TVM + // TVM EP takes the whole graph and optimizes it within its own framework. + // It does not retain the original graph. + ASSERT_EQ(0, model_proto.functions_size()); +#else ASSERT_EQ(1, model_proto.functions_size()); +#endif } } // namespace test