Function fft_c2r

src/backend/opencl/fft.cpp:129–155

template<typename Tr, typename Tc>
Array<Tr> fft_c2r(const Array<Tc> &in, const dim4 &odims, const int rank) {
    Array<Tr> out = createEmptyArray<Tr>(odims);

    verifySupported(rank, odims);
    size_t tdims[AF_MAX_DIMS], istrides[AF_MAX_DIMS], ostrides[AF_MAX_DIMS];

    computeDims(tdims, odims);
    computeDims(istrides, in.strides());
    computeDims(ostrides, out.strides());

    int batch = 1;
    for (int i = rank; i < AF_MAX_DIMS; i++) { batch *= tdims[i]; }

    SharedPlan plan = findPlan(
        CLFFT_HERMITIAN_INTERLEAVED, CLFFT_REAL, static_cast<clfftDim>(rank),
        tdims, istrides, istrides[rank], ostrides, ostrides[rank],
        static_cast<clfftPrecision>(Precision<Tc>::type), batch);

    cl_mem imem            = (*in.get())();
    cl_mem omem            = (*out.get())();
    cl_command_queue queue = getQueue()();

    CLFFT_CHECK(clfftEnqueueTransform(*plan.get(), CLFFT_BACKWARD, 1, &queue, 0,
                                      NULL, NULL, &imem, &omem, NULL));

    return out;
}

#define INSTANTIATE(T) \
    template void fft_inplace<T>(Array<T> &, const int, const bool);