mirrors/faiss
1
0
Fork 0
mirror of https://github.com/facebookresearch/faiss.git synced 2025-06-03 21:54:02 +08:00
Code Issues Projects Releases Wiki Activity

Include IndexResidual in clone_index

Browse Source 
Summary:
Moving index from cpu to gpu is failing with error message `RuntimeError: Error in virtual faiss::Index *faiss::Cloner::clone_Index(const faiss::Index *) at faiss/clone_index.cpp:144: clone not supported for this type of Index`
This diff support IndexResidual clone and unblocks gpu training

Reviewed By: sc268, mdouze

Differential Revision: D28614996

fbshipit-source-id: 9b1e5e7c5dd5da6d55f02594b062691565a86f49
This commit is contained in:
Sugosh Nagavara Ravindra 2021-05-24 20:04:00 -07:00 committed by Facebook GitHub Bot
parent d5a1bf3e9b
commit 0825eaf8d3
1 changed files with 2 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
faiss/clone_index.cpp
View File

@ -27,6 +27,7 @@
#include <faiss/IndexNSG.h>
#include <faiss/IndexPQ.h>
#include <faiss/IndexPreTransform.h>
#include <faiss/IndexResidual.h>
#include <faiss/IndexScalarQuantizer.h>
#include <faiss/MetaIndexes.h>
#include <faiss/VectorTransform.h>
@ -79,6 +80,7 @@ Index* Cloner::clone_Index(const Index* index) {
TRYCLONE(IndexFlatIP, index)
TRYCLONE(IndexFlat, index)
TRYCLONE(IndexLattice, index)
TRYCLONE(IndexResidual, index)
TRYCLONE(IndexScalarQuantizer, index)
TRYCLONE(MultiIndexQuantizer, index)
if (const IndexIVF* ivf = dynamic_cast<const IndexIVF*>(index)) {