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

Merge branch 'fix_testivf' into 'branch-0.5.0' 

MS-648 fix ivf test

See merge request 

Former-commit-id: 1f623de80f8e4c2668f2a30c7a069433c9d333a6
pull/191/head
peng.xu 2019-10-14 16:00:11 +08:00
parent e037b64d65 6f74ca8260
commit 25894b28ee
1 changed files with 2 additions and 6 deletions
core/src/index/knowhere/knowhere/index/vector_index

8
core/src/index/knowhere/knowhere/index/vector_index/IndexIVFSQHybrid.cpp
View File

@ -79,12 +79,8 @@ IVFSQHybrid::CopyGpuToCpu(const Config& config) {
VectorIndexPtr VectorIndexPtr
IVFSQHybrid::CopyCpuToGpu(const int64_t& device_id, const Config& config) { IVFSQHybrid::CopyCpuToGpu(const int64_t& device_id, const Config& config) {
if (auto res = FaissGpuResourceMgr::GetInstance().GetRes(device_id)) { auto p = CopyCpuToGpuWithQuantizer(device_id, config);
auto p = CopyCpuToGpuWithQuantizer(device_id, config); return p.first;
return p.first;
} else {
KNOWHERE_THROW_MSG("CopyCpuToGpu Error, can't get gpu_resource");
}
} }
void void