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

[GPU] Remove batched key from config (#9451)

Browse Source
This commit is contained in:
Roman Lyamin 2021-12-29 09:58:48 +03:00 committed by GitHub
parent 8b6a065948
commit 9f69daf0f3
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 2 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
src/inference/include/ie/gpu/gpu_config.hpp
View File

@ -156,11 +156,10 @@ DECLARE_GPU_CONFIG_KEY(MAX_NUM_THREADS);
DECLARE_GPU_CONFIG_KEY(ENABLE_LOOP_UNROLLING);
/**
* @brief This keys instructs the GPU plugin to use surface/buffer and batched memory type.
* @brief These keys instruct the GPU plugin to use surface/buffer memory type.
*/
DECLARE_GPU_CONFIG_KEY(SURFACE);
DECLARE_GPU_CONFIG_KEY(BUFFER);
DECLARE_GPU_CONFIG_KEY(BATCHED);
} // namespace GPUConfigParams

2
src/tests/functional/plugin/gpu/remote_blob_tests/gpu_remote_tensor_tests.cpp
View File

@ -782,7 +782,7 @@ TEST_P(OVRemoteTensorBatched_Test, NV12toBGR_image) {
auto p = PrePostProcessor(fn_ptr_remote);
p.input().tensor().set_element_type(ov::element::u8)
.set_color_format(ov::preprocess::ColorFormat::NV12_TWO_PLANES, {"y", "uv"})
.set_memory_type(std::string(GPU_CONFIG_KEY(SURFACE)) + GPU_CONFIG_KEY(BATCHED));
.set_memory_type(GPU_CONFIG_KEY(SURFACE));
p.input().preprocess().convert_color(ov::preprocess::ColorFormat::BGR);
p.input().model().set_layout("NCHW");
auto function = p.build();