[GPU] fixed to create Graphs with different stream_ids (#20626)

* fixed to create Graphs with different stream_ids

* added num_streams config for auto batch test
This commit is contained in:
Eddy Kim 2023-10-25 12:26:36 +09:00 committed by GitHub
parent 5fee2ef67e
commit 6fa4f9fd78
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 3 additions and 1 deletions

View File

@ -169,7 +169,7 @@ CompiledModel::CompiledModel(cldnn::BinaryInputBuffer ib,
auto pos = ib.tellg();
for (uint16_t n = 0; n < m_config.get_property(ov::num_streams); n++) {
ib.seekg(pos);
auto graph = std::make_shared<Graph>(ib, context, m_config, 0);
auto graph = std::make_shared<Graph>(ib, context, m_config, n);
m_graphs.push_back(graph);
}
}

View File

@ -68,6 +68,8 @@ namespace {
{{CONFIG_KEY(AUTO_BATCH_DEVICE_CONFIG), ov::test::utils::DEVICE_GPU}},
{{CONFIG_KEY(AUTO_BATCH_DEVICE_CONFIG), ov::test::utils::DEVICE_GPU},
{CONFIG_KEY(AUTO_BATCH_TIMEOUT), "1"}},
{{CONFIG_KEY(AUTO_BATCH_DEVICE_CONFIG), ov::test::utils::DEVICE_GPU},
{ov::num_streams.name(), "AUTO"}},
};
};