未验证 提交 a7932b03 编写于 作者: C Cai Yudong 提交者: GitHub

#1781 fix search hang with SQ8H (#1783)

Signed-off-by: Nyudong.cai <yudong.cai@zilliz.com>
上级 af73dccb
...@@ -21,6 +21,7 @@ Please mark all change in change log and use the issue from GitHub ...@@ -21,6 +21,7 @@ Please mark all change in change log and use the issue from GitHub
- \#1735 Fix search out of memory with ivf_flat - \#1735 Fix search out of memory with ivf_flat
- \#1747 Expected error status if search with partition_tag not existed - \#1747 Expected error status if search with partition_tag not existed
- \#1756 Fix memory exhausted during searching - \#1756 Fix memory exhausted during searching
- \#1781 Fix search hang with SQ8H
## Feature ## Feature
- \#1603 BinaryFlat add 2 Metric: Substructure and Superstructure - \#1603 BinaryFlat add 2 Metric: Substructure and Superstructure
......
...@@ -551,11 +551,11 @@ ExecutionEngineImpl::CopyToGpu(uint64_t device_id, bool hybrid) { ...@@ -551,11 +551,11 @@ ExecutionEngineImpl::CopyToGpu(uint64_t device_id, bool hybrid) {
*/ */
ENGINE_LOG_DEBUG << "CPU to GPU" << device_id << " start"; ENGINE_LOG_DEBUG << "CPU to GPU" << device_id << " start";
auto gpu_cache_mgr = cache::GpuCacheMgr::GetInstance(device_id); auto gpu_cache_mgr = cache::GpuCacheMgr::GetInstance(device_id);
gpu_cache_mgr->Lock(); // gpu_cache_mgr->Lock();
gpu_cache_mgr->Reserve(index_->Size()); // gpu_cache_mgr->Reserve(index_->Size());
index_ = knowhere::cloner::CopyCpuToGpu(index_, device_id, knowhere::Config()); index_ = knowhere::cloner::CopyCpuToGpu(index_, device_id, knowhere::Config());
gpu_cache_mgr->InsertItem(location_, std::static_pointer_cast<cache::DataObj>(index_)); // gpu_cache_mgr->InsertItem(location_, std::static_pointer_cast<cache::DataObj>(index_));
gpu_cache_mgr->Unlock(); // gpu_cache_mgr->Unlock();
ENGINE_LOG_DEBUG << "CPU to GPU" << device_id << " finished"; ENGINE_LOG_DEBUG << "CPU to GPU" << device_id << " finished";
} catch (std::exception& e) { } catch (std::exception& e) {
ENGINE_LOG_ERROR << e.what(); ENGINE_LOG_ERROR << e.what();
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册