Open
Description
A 1d fft across the nth dimension of > n dimensional CuArray (where n is not 1) is not enabled by the wrapper (ERROR: ArgumentError: batching dims must be sequential from CuArrays>wXQp8>src>fft>wrappers.jl)
to reproduce:
using CuArrays, CUDAnative, CUDAdrv, FFTW
dim = 2
data = CuArrays.rand(ComplexF32,512,512,512);
myFft = plan_fft!(data,dim);
the wrapper for regular cpu arrays allows this, and if dim is 1 or 3 it also works as expected
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment