Restore IVF-SQ list size initialization

Author: bdice

c/tests/neighbors/ann_mg_c.cu

@@ -6,6 +6,7 @@
 #include <cuda.h>
 #include <gtest/gtest.h>
 #include <cuda/stream>
+
 #include <raft/core/device_mdarray.hpp>
 #include <raft/core/handle.hpp>
 #include <raft/random/rng.cuh>

cpp/src/neighbors/ivf_sq_index.cpp

@@ -46,6 +46,8 @@ index<CodeT>::index(raft::resources const& res,
   check_consistency();
   auto stream = raft::resource::get_cuda_stream(res);
   std::memset(accum_sorted_sizes_.data_handle(), 0, accum_sorted_sizes_.size() * sizeof(int64_t));
+  RAFT_CUDA_TRY(cudaMemsetAsync(
+    list_sizes_.data_handle(), 0, list_sizes_.size() * sizeof(uint32_t), stream.get()));
   RAFT_CUDA_TRY(
     cudaMemsetAsync(data_ptrs_.data_handle(), 0, data_ptrs_.size() * sizeof(CodeT*), stream.get()));
   RAFT_CUDA_TRY(cudaMemsetAsync(