diff --git a/src/plugins/auto/src/executable_network.cpp b/src/plugins/auto/src/executable_network.cpp index e732a1fb5e7..c85c49e5801 100644 --- a/src/plugins/auto/src/executable_network.cpp +++ b/src/plugins/auto/src/executable_network.cpp @@ -34,4 +34,11 @@ void ExecutableNetwork::SetExeNetworkForContext() { std::string ExecutableNetwork::GetLogTag() const noexcept { return _sContext->_LogTag; } + +std::shared_ptr ExecutableNetwork::GetExecGraphInfo() { + // TODO: When there are more then 2 executable networks, such as MULTI:GPU,CPU + auto autoSContext = std::dynamic_pointer_cast(_sContext); + return autoSContext->_hwExecutableNetwork->GetExecGraphInfo(); +} + } // namespace MultiDevicePlugin diff --git a/src/plugins/auto/src/executable_network.hpp b/src/plugins/auto/src/executable_network.hpp index b52aaca273d..d0ce4f48780 100644 --- a/src/plugins/auto/src/executable_network.hpp +++ b/src/plugins/auto/src/executable_network.hpp @@ -20,6 +20,7 @@ public: using Ptr = std::shared_ptr; ExecutableNetwork(const Schedule::Ptr& schedule, const ScheduleContext::Ptr& sContext); IInferPtr CreateInferRequest() override; + std::shared_ptr GetExecGraphInfo() override; ~ExecutableNetwork() override; protected: diff --git a/src/tests/functional/plugin/shared/src/behavior/executable_network/exec_graph_info.cpp b/src/tests/functional/plugin/shared/src/behavior/executable_network/exec_graph_info.cpp index 813f3108679..c425887f30e 100644 --- a/src/tests/functional/plugin/shared/src/behavior/executable_network/exec_graph_info.cpp +++ b/src/tests/functional/plugin/shared/src/behavior/executable_network/exec_graph_info.cpp @@ -424,7 +424,11 @@ TEST_P(ExecGraphSerializationTest, ExecutionGraph) { pugi::xml_document expected; pugi::xml_document result; - ASSERT_TRUE(expected.load_string(target_device == "CPU" ? expected_serialized_model_cpu : expected_serialized_model)); + if (target_device == "CPU" || target_device == "AUTO:CPU" || target_device == "MULTI:CPU") { + ASSERT_TRUE(expected.load_string(expected_serialized_model_cpu)); + } else { + ASSERT_TRUE(expected.load_string(expected_serialized_model)); + } ASSERT_TRUE(result.load_file(m_out_xml_path.c_str())); bool status;