Commit 1dcb2799 authored by 王翔宇's avatar 王翔宇
Browse files

update


Former-commit-id: 25e3afb1d07e41b47b7c6f1b60c45a4e431f29fe
parent b6cbacb9
Loading
Loading
Loading
Loading
+8 −0
Original line number Diff line number Diff line
@@ -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();
}