diff --git a/onnxruntime/test/framework/session_state_test.cc b/onnxruntime/test/framework/session_state_test.cc index 533f7792d5..e7f8b1aaa4 100644 --- a/onnxruntime/test/framework/session_state_test.cc +++ b/onnxruntime/test/framework/session_state_test.cc @@ -347,7 +347,7 @@ TEST(SessionStateTest, TestInitializerMemoryAllocatedUsingNonArenaMemory) { SessionState session_state(graph, execution_providers, nullptr, nullptr, dtm, edlm, DefaultLoggingManager().DefaultLogger(), profiler, sess_options); - + // Partition the graph GraphPartitioner partitioner(krm, execution_providers); ASSERT_STATUS_OK(partitioner.Partition( @@ -406,7 +406,7 @@ TEST(SessionStateTest, TestInitializerMemoryAllocatedUsingNonArenaMemory) { SessionState session_state(graph, execution_providers, nullptr, nullptr, dtm, edlm, DefaultLoggingManager().DefaultLogger(), profiler, sess_options); - + // Partition the graph GraphPartitioner partitioner(krm, execution_providers); ASSERT_STATUS_OK(partitioner.Partition(