admin/milvus
1
0
Fork 0
mirror of https://gitee.com/milvus-io/milvus.git synced 2026-01-07 19:31:51 +08:00
Code Issues Packages Projects Releases Wiki Activity

Server crashed when searching vectors using GPU(#1429) (#1458)

Browse Source 
Signed-off-by: groot <yihua.mo@zilliz.com>
This commit is contained in:
groot 2020-03-01 15:51:28 +08:00 committed by GitHub
parent d380c8d69b
commit 220ecf5fac
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 16 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
CHANGELOG.md
View File

@ -27,6 +27,7 @@ Please mark all change in change log and use the issue from GitHub
- \#1240 - Update license declaration
- \#1298 - Unittest failed when on CPU2GPU case
- \#1359 - Negative distance value returned when searching with HNSW index type
- \#1429 - Server crashed when searching vectors using GPU
## Feature
- \#216 - Add CLI to get server info

21
core/src/index/knowhere/knowhere/index/vector_index/helpers/Cloner.cpp
View File

@ -32,27 +32,36 @@ CopyGpuToCpu(const VectorIndexPtr& index, const Config& config) {
VectorIndexPtr
CopyCpuToGpu(const VectorIndexPtr& index, const int64_t& device_id, const Config& config) {
VectorIndexPtr result;
auto uids = index->GetUids();
#ifdef CUSTOMIZATION
if (auto device_index = std::dynamic_pointer_cast<IVFSQHybrid>(index)) {
return device_index->CopyCpuToGpu(device_id, config);
result = device_index->CopyCpuToGpu(device_id, config);
result->SetUids(uids);
return result;
}
#endif
if (auto device_index = std::dynamic_pointer_cast<GPUIndex>(index)) {
return device_index->CopyGpuToGpu(device_id, config);
result = device_index->CopyGpuToGpu(device_id, config);
result->SetUids(uids);
return result;
}
if (auto cpu_index = std::dynamic_pointer_cast<IVFSQ>(index)) {
return cpu_index->CopyCpuToGpu(device_id, config);
result = cpu_index->CopyCpuToGpu(device_id, config);
} else if (auto cpu_index = std::dynamic_pointer_cast<IVFPQ>(index)) {
return cpu_index->CopyCpuToGpu(device_id, config);
result = cpu_index->CopyCpuToGpu(device_id, config);
} else if (auto cpu_index = std::dynamic_pointer_cast<IVF>(index)) {
return cpu_index->CopyCpuToGpu(device_id, config);
result = cpu_index->CopyCpuToGpu(device_id, config);
} else if (auto cpu_index = std::dynamic_pointer_cast<IDMAP>(index)) {
return cpu_index->CopyCpuToGpu(device_id, config);
result = cpu_index->CopyCpuToGpu(device_id, config);
} else {
KNOWHERE_THROW_MSG("this index type not support tranfer to gpu");
}
result->SetUids(uids);
return result;
}
} // namespace cloner