search

lebedov / scikit-cuda

Python interface to GPU-powered libraries
http://scikit-cuda.readthedocs.org/
Other
987 stars 179 forks source link

CUDA Streams #10

Closed jtksai closed 12 years ago

jtksai commented 13 years ago

Is it possible to use CUDA streams with the fft functions? Nvidia's CUFFT manual clearly says yes. However I haven't found any mention of streams or cufftSetStream function in the fft function python code. I'm writing/testing a pseudo-spectral PDE solver and stream could potentially speed things a bit.

lebedov commented 13 years ago

Currently, no. PyCUDA provides an interface to CUDA's stream features, but doesn't expose the actual stream identifier that would need to be passed to cufftSetStream. I'll ask the PyCUDA developer whether it would be possible to expose the identifier.

lebedov commented 13 years ago

A means of accessing the identifier has been added to the development version of PyCUDA. I'll check whether it can be used to pass streams to the CUFFT and CUBLAS functions.

lebedov commented 12 years ago

Streams can now be specified when creating a plan.