Fix setting the correct stream for cufft

Description

This is needed in case one has more than one batch per core as the stream is reset for each batch.

Checklist

• I have checked that my code follows the Octopus coding standards
• I have added tests for all the new features added in this request.