add check for .enc extension to support encryption (#467)

* add check for .enc extension to support encryption * check rotation_matrix file in file blobs
microsoft · Oct 4, 2023 · aa6e33a · aa6e33a
1 parent fbcdb4b
commit aa6e33a
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/src/pq.cpp b/src/pq.cpp
@@ -133,11 +133,13 @@ void FixedChunkPQTable::load_pq_centroid_bin(const char *pq_table_file, size_t n
     diskann::cout << "Loaded PQ Pivots: #ctrs: " << NUM_PQ_CENTROIDS << ", #dims: " << this->ndims
                   << ", #chunks: " << this->n_chunks << std::endl;
 
-    if (file_exists(rotmat_file))
-    {
 #ifdef EXEC_ENV_OLS
+    if (files.fileExists(rotmat_file))
+    {
         diskann::load_bin<float>(files, rotmat_file, (float *&)rotmat_tr, nr, nc);
 #else
+    if (file_exists(rotmat_file))
+    {
         diskann::load_bin<float>(rotmat_file, rotmat_tr, nr, nc);
 #endif
         if (nr != this->ndims || nc != this->ndims)