search

vincefn / pyvkfft

Python interface to VkFFT
MIT License
51 stars 6 forks source link

Batched transforms not working (OpenCL) #5

Closed memmett closed 3 years ago

memmett commented 3 years ago

Hello, it appears as though batched transforms are not working correctly (or I have misunderstood the documentation). Please find a simple example attached.

In the example, I use vkFFT to do a forward complex-to-complex 1D batched transform and compare the result to NumPy FFT. The relative different is O(1). If I instead do the same transform, but one batch at a time, then the vkFFT/NumPy transforms match exactly.

I will try to find a solution, but you might know exactly where to look!

Thanks, Matt

vkfft_batch.zip

memmett commented 3 years ago

Yep, my bad. I was passing dim instead of ndim!