Multi-gpu support, correct cuda stream usage in simple fft interface

vincefn / pyvkfft

Python interface to VkFFT

MIT License

51 stars 6 forks source link

Closed vincefn closed 1 year ago

vincefn commented 1 year ago

Take into account the current cuda device when automatically caching the VkFFTApp using the pyvkfft.fft interface (closes #26).
Actually use the cuda_stream parameter in the pyvkfft.fft interface.
Prevent F-ordered inplace R2C tests with cupy