saymrwulf/onnxruntime
1
0
Fork 0
mirror of https://github.com/saymrwulf/onnxruntime.git synced 2026-06-28 03:20:58 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Fix unit test when TVM EP is enabled (#18189)

Browse source 
### Description

TestInlinedLocalFunctionNotRemoved checks that local functions are not
removed but TVM EP optimizes the whole graph after it is inlined.
This commit is contained in:
Xavier Dupré 2023-11-06 19:32:26 +01:00 committed by GitHub
parent 398ef677ba
commit 3b63d85c25
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
1 changed files with 6 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

6
onnxruntime/test/framework/function_test.cc
View file

@ -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