Former-commit-id: 99939385bdc36200807099b56fc49d72578489f1
pull/191/head
wxyu 2019-10-19 19:33:06 +08:00
parent 964e1d5cfd
commit ac20f4f9a6
1 changed files with 8 additions and 0 deletions

View File

@ -124,6 +124,11 @@ ExecutionEngineImpl::HybridLoad() const {
return;
}
if (index_->GetType() == IndexType::FAISS_IDMAP) {
ENGINE_LOG_WARNING << "HybridLoad with type FAISS_IDMAP, ignore";
return;
}
const std::string key = location_ + ".quantizer";
std::vector<uint64_t> gpus = scheduler::get_gpu_pool();
@ -178,6 +183,9 @@ ExecutionEngineImpl::HybridUnset() const {
if (index_type_ != EngineType::FAISS_IVFSQ8H) {
return;
}
if (index_->GetType() == IndexType::FAISS_IDMAP) {
return;
}
index_->UnsetQuantizer();
}