We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent 8b35dfa commit 7d83232Copy full SHA for 7d83232
1 file changed
faiss/IndexIVF.cpp
@@ -91,6 +91,9 @@ void Level1Quantizer::train_q1(
91
} else {
92
clus.train(n, x, *quantizer);
93
}
94
+#ifdef __aarch64__
95
+ quantizer->train(-1, x);
96
+#endif
97
quantizer->is_trained = true;
98
} else if (quantizer_trains_alone == 2) {
99
if (verbose) {
@@ -471,7 +474,9 @@ void IndexIVF::search_preassigned(
471
474
get_InvertedListScanner(store_pairs, sel, params));
472
475
473
476
#ifdef __aarch64__
- krl_create_LUT8b_handle(&(scanner->klh),(int)(sel != nullptr), tmp_buffer_size);
477
+ if (tmp_buffer_size > 0) {
478
+ krl_create_LUT8b_handle(&(scanner->klh),(int)(sel != nullptr), tmp_buffer_size);
479
+ }
480
#endif
481
/*****************************************************
482
* Depending on parallel_mode, there are two possible ways
0 commit comments