Commit 0527b46
Update __init__.py (facebookresearch#4086)
Summary:
X-link: meta-pytorch/torchrec#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.
Pull Request resolved: facebookresearch#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: aa05aa20fe61e4ad75d4234f52e3b0d6a673a0b91 parent 55bab92 commit 0527b46
1 file changed
Lines changed: 5 additions & 3 deletions
| Original file line number | Diff line number | Diff line change | |
|---|---|---|---|
| |||
9 | 9 | | |
10 | 10 | | |
11 | 11 | | |
| 12 | + | |
12 | 13 | | |
13 | 14 | | |
14 | 15 | | |
| |||
30 | 31 | | |
31 | 32 | | |
32 | 33 | | |
| 34 | + | |
| 35 | + | |
33 | 36 | | |
34 | 37 | | |
35 | 38 | | |
| |||
161 | 164 | | |
162 | 165 | | |
163 | 166 | | |
164 | | - | |
165 | 167 | | |
166 | 168 | | |
167 | 169 | | |
168 | 170 | | |
169 | 171 | | |
170 | | - | |
171 | | - | |
| 172 | + | |
| 173 | + | |
172 | 174 | | |
173 | 175 | | |
174 | 176 | | |
| |||
0 commit comments