Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Update __init__.py #4086

Closed
wants to merge 2 commits into from
Closed

Update __init__.py #4086

wants to merge 2 commits into from

Conversation

asadoughi
Copy link
Contributor

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.

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
@asadoughi asadoughi changed the title Asadoughi patch 1 Update __init__.py Dec 12, 2024
@facebook-github-bot
Copy link
Contributor

@asadoughi has imported this pull request. If you are a Meta employee, you can view this diff on Phabricator.

asadoughi added a commit to asadoughi/torchrec that referenced this pull request Dec 12, 2024
Summary:
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

Reviewed By: mnorris11

Differential Revision: D67118292

Pulled By: asadoughi
@facebook-github-bot
Copy link
Contributor

@asadoughi merged this pull request in 5f0a99a.

facebook-github-bot pushed a commit to pytorch/torchrec that referenced this pull request Dec 12, 2024
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
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Projects
None yet
Development

Successfully merging this pull request may close these issues.

3 participants