facebookresearch / facebookresearch/faiss

cudaStream_t arguments in Python client

Open
#656 4 comments 0 reactions 0 assignees View on GitHub
feature request GPU
Dominant language
C++
Stars
40.9k
Forks
4.5k
PR merge metrics
No merged PRs in 30d

Description

The Python SWIG client exposes the setDefaultStream() function on the GPU resources object but it does not seem to provide any type conversion options to pass in a Python equivalent to the C++ cudastream_t.

Is there a recommended way to accomplish this?

```python
>>> res.setDefaultStream(0, 0)
Traceback (most recent call last):
File "", line 1, in
File "/raid/cjnolet/dev_env/lib/python3.5/site-packages/faiss-0.1-py3.5.egg/faiss/swigfaiss_gpu.py", line 1193, in setDefaultStream
return _swigfaiss_gpu.StandardGpuResources_setDefaultStream(self, device, stream)
TypeError: in method 'StandardGpuResources_setDefaultStream', argument 3 of type 'cudaStream_t'
```

Running on:
- [ ] CPU
- [X] GPU

Interface:
- [ ] C++
- [x] Python

Contributor guide

Open the contributing guide

Assessment

This issue has not been assessed yet.

Get new issues in your inbox

A short digest of beginner-friendly GitHub issues.