heffte_compute_transform.h

/*
    -- heFFTe --
       Univ. of Tennessee, Knoxville
       @date
*/
 
#ifndef HEFFTE_COMPUTE_TRANSFORM_H
#define HEFFTE_COMPUTE_TRANSFORM_H
 
#include "heffte_reshape3d.h"
 
namespace heffte {
 
    template<typename location_tag, typename index, typename scalar_type>
    void compute_transform(typename backend::data_manipulator<location_tag>::stream_type stream,
                        int const batch_size,
                        scalar_type const input[], scalar_type output[], scalar_type workspace[],
                        size_t executor_buffer_offset, size_t size_comm_buffers,
                        std::array<std::unique_ptr<reshape3d_base<index>>, 4> const &shaper,
                        std::array<executor_base*, 3> const &executor,
                        direction dir);
    template<typename location_tag, typename index, typename scalar_type>
    void compute_transform(typename backend::data_manipulator<location_tag>::stream_type stream,
                        int const batch_size,
                        scalar_type const input[], std::complex<scalar_type> output[],
                        std::complex<scalar_type> workspace[],
                        size_t executor_buffer_offset, size_t size_comm_buffers,
                        std::array<std::unique_ptr<reshape3d_base<index>>, 4> const &shaper,
                        std::array<executor_base*, 3> const &executor, direction);
    template<typename location_tag, typename index, typename scalar_type>
    void compute_transform(typename backend::data_manipulator<location_tag>::stream_type stream,
                        int const batch_size,
                        std::complex<scalar_type> const input[], scalar_type output[],
                        std::complex<scalar_type> workspace[],
                        size_t executor_buffer_offset, size_t size_comm_buffers,
                        std::array<std::unique_ptr<reshape3d_base<index>>, 4> const &shaper,
                        std::array<executor_base*, 3> const &executor, direction);
 
}
 
#endif // HEFFTE_COMPUTE_TRANSFORM_H