natas_hw
/
openvino
0
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

[GPU] Fix gpu dynamic model multistream test issue (#16510) (#16510)

This commit is contained in:
Paul Youngsoo Ahn 2023-03-24 02:51:57 +09:00 committed by GitHub
parent 2755b32fb9
commit 74870f9b0b
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 3 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
src/plugins/intel_gpu/tests/test_cases/multiple_streams_gpu_test.cpp
View File

@ -48,6 +48,8 @@ TEST(multistream_gpu, basic) {
std::vector<InferenceEngine::Task> tasks;
for (size_t i = 0; i < num_streams; i++) {
tasks.push_back([&networks, i, &engine] {
auto cfg = get_test_default_config(engine);
auto stream = engine.create_stream(cfg);
auto net = networks[i];
std::vector<int> various_size = {32, 128, 16, 64};
for (size_t iter = 0; iter < 8; iter++) {
@ -60,7 +62,7 @@ TEST(multistream_gpu, basic) {
auto outputs = net->execute();
auto output = outputs.at("shape_of").get_memory();
cldnn::mem_lock<int32_t> output_ptr(output, get_test_stream());
cldnn::mem_lock<int32_t> output_ptr(output, *stream);
std::vector<int32_t> expected_results = {1, len, 512};