Commit 8a2d5c82 authored by xiaojun.lin's avatar xiaojun.lin
Browse files

fix copy_cpu_to_gpu


Former-commit-id: 8703a392ac93bd9450492a6841a169bb2287c6ea
parent 25894b28
Loading
Loading
Loading
Loading
+18 −2
Original line number Diff line number Diff line
@@ -79,8 +79,24 @@ IVFSQHybrid::CopyGpuToCpu(const Config& config) {

VectorIndexPtr
IVFSQHybrid::CopyCpuToGpu(const int64_t& device_id, const Config& config) {
    auto p = CopyCpuToGpuWithQuantizer(device_id, config);
    return p.first;
    if (auto res = FaissGpuResourceMgr::GetInstance().GetRes(device_id)) {
        ResScope rs(res, device_id, false);
        faiss::gpu::GpuClonerOptions option;
        option.allInGpu = true;

        faiss::IndexComposition index_composition;
        index_composition.index = index_.get();
        index_composition.quantizer = nullptr;
        index_composition.mode = 0;  // copy all

        auto gpu_index = faiss::gpu::index_cpu_to_gpu(res->faiss_res.get(), device_id, &index_composition, &option);

        std::shared_ptr<faiss::Index> device_index = std::shared_ptr<faiss::Index>(gpu_index);;
        auto new_idx = std::make_shared<IVFSQHybrid>(device_index, device_id, res);
        return new_idx;
    } else {
        KNOWHERE_THROW_MSG("CopyCpuToGpu Error, can't get gpu_resource");
    }
}

void