milvus-io
/
milvus
mirror of https://github.com/milvus-io/milvus.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

add log 

Former-commit-id: 12f5c2eb2903b8285c2d4ae3fb58e02fd10c3f83
pull/191/head
wxyu 2019-10-19 15:03:40 +08:00
parent d72d6fef48
commit b8605caa40
2 changed files with 5 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
core/src/db/engine/ExecutionEngineImpl.cpp
View File

@ -164,6 +164,9 @@ ExecutionEngineImpl::HybridLoad() const {
quantizer_conf->mode = 1; quantizer_conf->mode = 1;
quantizer_conf->gpu_id = best_device_id; quantizer_conf->gpu_id = best_device_id;
auto quantizer = index_->LoadQuantizer(quantizer_conf); auto quantizer = index_->LoadQuantizer(quantizer_conf);
if (quantizer == nullptr) {
ENGINE_LOG_ERROR << "quantizer is nullptr";
}
index_->SetQuantizer(quantizer); index_->SetQuantizer(quantizer);
auto cache_quantizer = std::make_shared<CachedQuantizer>(quantizer); auto cache_quantizer = std::make_shared<CachedQuantizer>(quantizer);
cache::GpuCacheMgr::GetInstance(best_device_id)->InsertItem(key, cache_quantizer); cache::GpuCacheMgr::GetInstance(best_device_id)->InsertItem(key, cache_quantizer);

2
core/src/wrapper/VecIndex.h
View File

@ -26,6 +26,7 @@
#include "knowhere/common/Config.h" #include "knowhere/common/Config.h"
#include "knowhere/index/vector_index/Quantizer.h" #include "knowhere/index/vector_index/Quantizer.h"
#include "utils/Status.h" #include "utils/Status.h"
#include "utils/Log.h"
namespace milvus { namespace milvus {
namespace engine { namespace engine {
@ -101,6 +102,7 @@ class VecIndex : public cache::DataObj {
//////////////// ////////////////
virtual knowhere::QuantizerPtr virtual knowhere::QuantizerPtr
LoadQuantizer(const Config& conf) { LoadQuantizer(const Config& conf) {
ENGINE_LOG_ERROR << "LoadQuantizer virtual funciton called.";
return nullptr; return nullptr;
} }