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

[GPU] Fix va surface sharing issue (#20730)

Browse Source
This commit is contained in:
Wilson Seok 2023-10-30 21:27:50 +09:00 committed by GitHub
parent c3a90f8f70
commit 6dce6cc599
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 5 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
src/plugins/intel_gpu/src/plugin/remote_tensor.cpp
View File

@ -109,6 +109,10 @@ void RemoteTensorImpl::allocate() {
auto context = std::dynamic_pointer_cast<RemoteContextImpl>(m_context); auto context = std::dynamic_pointer_cast<RemoteContextImpl>(m_context);
auto enable_caching = supports_caching(); auto enable_caching = supports_caching();
if (is_surface()) {
m_layout.format = cldnn::format::nv12; // Other formats are not supported
}
if (enable_caching) { if (enable_caching) {
m_memory_object = context->try_get_cached_memory(m_hash); m_memory_object = context->try_get_cached_memory(m_hash);
if (m_memory_object) { if (m_memory_object) {
@ -156,7 +160,6 @@ void RemoteTensorImpl::allocate() {
} }
#ifdef _WIN32 #ifdef _WIN32
case TensorType::BT_SURF_SHARED: { case TensorType::BT_SURF_SHARED: {
m_layout.format = cldnn::format::nv12; // Other formats are not supported
m_memory_object = engine.share_surface(m_layout, m_mem, m_plane); m_memory_object = engine.share_surface(m_layout, m_mem, m_plane);
break; break;
} }
@ -166,13 +169,11 @@ void RemoteTensorImpl::allocate() {
} }
#else #else
case TensorType::BT_SURF_SHARED: { case TensorType::BT_SURF_SHARED: {
m_layout.format = cldnn::format::nv12; // Other formats are not supported
m_memory_object = engine.share_surface(m_layout, m_surf, m_plane); m_memory_object = engine.share_surface(m_layout, m_surf, m_plane);
break; break;
} }
#endif #endif
case TensorType::BT_IMG_SHARED: { case TensorType::BT_IMG_SHARED: {
m_layout.format = cldnn::format::nv12; // Other formats are not supported
m_memory_object = engine.share_image(m_layout, m_mem); m_memory_object = engine.share_image(m_layout, m_mem);
break; break;
} }
@ -218,8 +219,7 @@ void RemoteTensorImpl::update_hash() {
bool RemoteTensorImpl::is_surface() const noexcept { bool RemoteTensorImpl::is_surface() const noexcept {
return m_mem_type == TensorType::BT_SURF_SHARED || return m_mem_type == TensorType::BT_SURF_SHARED ||
m_mem_type == TensorType::BT_IMG_SHARED || m_mem_type == TensorType::BT_IMG_SHARED;
m_mem_type == TensorType::BT_DX_BUF_SHARED;
} }
cldnn::memory::ptr RemoteTensorImpl::get_memory() const { cldnn::memory::ptr RemoteTensorImpl::get_memory() const {