Merge pull request facebookresearch#1006 from milvus-io/1.6.0

wickedfoo · web-flow · commit 2cce2e5f59a5 · 2019-10-31T23:57:57.000-04:00
Fix GPUSQ copyto bug
diff --git a/gpu/GpuIndexIVFScalarQuantizer.cu b/gpu/GpuIndexIVFScalarQuantizer.cu
@@ -139,6 +139,7 @@ GpuIndexIVFScalarQuantizer::copyTo(
 
   GpuIndexIVF::copyTo(index);
   index->sq = sq;
+  index->code_size = sq.code_size;
   index->by_residual = by_residual;
 
   InvertedLists* ivf = new ArrayInvertedLists(nlist, index->code_size);
