facebookresearch / facebookresearch/faiss
cudaStream_t arguments in Python client
- 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
Assessment
This issue has not been assessed yet.