[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 authored and GitHub committed 2023-10-24 20:26:36 -07:00
1 parent 5fee2ef67e
commit 6fa4f9fd78
2 files changed
+3 -1

No files matched your search

@@ -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);
}
}
@@ -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"}},
};
};