Skip to content

Commit 8afe20e

Browse files
asadoughifacebook-github-bot
authored andcommitted
Update __init__.py (#2628)
Summary: Pull Request resolved: #2628 Fix erroneous logging `Failed to load GPU Faiss: No module named 'swigfaiss_gpu'. Will not load constructor refs for GPU indexes.` when I was running on a GPU-based index just fine. `GpuIndex...` is loaded via loader.py. X-link: facebookresearch/faiss#4086 Test Plan: before ``` (faiss_gpu_cuvs_nightly) [[email protected] ~]$ python3 script.py Failed to load GPU Faiss: No module named 'swigfaiss_gpu'. Will not load constructor refs for GPU indexes. ``` after ``` (faiss_gpu_cuvs_nightly) [[email protected] ~]$ python3 Python 3.12.8 | packaged by conda-forge | (main, Dec 5 2024, 14:24:40) [GCC 13.3.0] on linux Type "help", "copyright", "credits" or "license" for more information. >>> import faiss >>> ``` Reviewed By: mnorris11 Differential Revision: D67118292 Pulled By: asadoughi fbshipit-source-id: aa05aa20fe61e4ad75d4234f52e3b0d6a673a0b9
1 parent 575e081 commit 8afe20e

File tree

1 file changed

+1
-0
lines changed

1 file changed

+1
-0
lines changed

Diff for: examples/retrieval/knn_index.py

+1
Original file line numberDiff line numberDiff line change
@@ -43,6 +43,7 @@ def get_index(
4343
res = faiss.StandardGpuResources()
4444
# pyre-fixme[16]
4545
config = faiss.GpuIndexIVFPQConfig()
46+
# pyre-ignore[16]
4647
index = faiss.GpuIndexIVFPQ(
4748
res,
4849
embedding_dim,

0 commit comments

Comments
 (0)