kml_fft(f/h)_mpi_plan_many_dft

建立howmany组数据序列n维C2C变换的plan。其中单个FFT的数据序列不需要是连续的，可以以跨步的形式提供。

接口定义

C interface：

kml_fft_plan kml_fft_mpi_plan_dft_3d(int rank, const int *n, int howmany, ptrdiff_t iblock, ptrdiff_t oblock, kml_fft_complex *in, kml_fft_complex *out, MPI_comm comm, int sign, unsigned flags);

kml_fftf_plan kml_fftf_mpi_plan_dft_3d(int rank, const int *n, int howmany, ptrdiff_t iblock, ptrdiff_t oblock, kml_fftf_complex *in, kml_fftf_complex *out, MPI_comm comm, int sign, unsigned flags);

kml_ffth_plan kml_ffth_mpi_plan_dft_3d(int rank, const int *n, int howmany, ptrdiff_t iblock, ptrdiff_t oblock, kml_ffth_complex *in, kml_ffth_complex *out, MPI_comm comm, int sign, unsigned flags);

返回值

函数返回一个kml_fft(f)_plan类型的结构体指针。将该对象作为参数传入kml_fft(f/h)_execute函数中使用，将对当前提供的输入in和输出out执行FFT变换；另外，也可以通过将该对象作为参数传入kml_fft(f/h)_execute_dft函数中以对新的输入in和输出out执行FFT变换。

如果函数返回非空指针，则表示plan执行成功，否则表示执行失败。

参数

参数名

数据类型

描述

输入/输出

rank

int

FFT变换的维度是rank，约束：1≤rank≤3。

输入

n

const int*

n是维度为rank的数组，包含FFT序列每一维度的大小，约束：n[i]≥1, for i in 0, rank - 1。

输入

howmany

int

howmany表示要多少个多维FFT变换。

输入

iblock

ptrdiff_t

n0维度上的block大小

输入

oblock

ptrdiff_t

n1维度上的block大小

输入

in
  • 双精度：kml_fft_complex*
  • 单精度：kml_fftf_complex*
  • 半精度：kml_ffth_complex*

输入待变换的数据。

输入

out
  • 双精度：kml_fft_complex*
  • 单精度：kml_fftf_complex*
  • 半精度：kml_ffth_complex*

输出快速傅里叶变换后的数据。

输出

comm

MPI_Comm

MPI通信器句柄

输入

sign

int

描述正向变换或反向变换。

  • -1(KML_FFT_FORWARD)：forward
  • +1(KML_FFT_BACKWARD)：backward

输入

flags

unsigned int

planning选项，未使用。

输入

依赖

C: "kfft-mpi.h"

示例

C interface：

    const ptrdiff_t N0 = 4, N1 = 4;
    kml_fft_plan plan;
    ptrdiff_t alloc_local, local_n0, local_0_start, i, j;
 
    MPI_Init(&argc, &argv);
    MPI_Comm comm = MPI_COMM_WORLD;
    kml_fft_mpi_init();
    kml_fft_complex *in = NULL;
    kml_fft_complex *out = NULL;
    ptrdiff_t block0 = 0;
    ptrdiff_t block1 = 0;
    ptrdiff_t howmany = 1;
    ptrdiff_t rank = 2;
    ptrdiff_t *n = (ptrdiff_t *)kml_fft_malloc(sizeof(ptrdiff_t) * rank);
    
    n[0] = N0;
    n[1] = N1;
 
    /* get local data size and allocate */
    alloc_local = kml_fft_mpi_local_size_many(rank, n, howmany, block0, comm,
                                         &local_n0, &local_0_start);
    if (alloc_local == -1) {
            printf("[%s][%d] allocate size fail!!!\n", __func__, __LINE__);
    }
    in = (kml_fft_complex *)kml_fft_malloc(sizeof(kml_fft_complex) * alloc_local);
    if (in == NULL) {
        printf("[%s][%d] malloc memory fail!!!\n", __func__, __LINE__);
    }
    out = (kml_fft_complex *)kml_fft_malloc(sizeof(kml_fft_complex) * alloc_local);
    if (out == NULL) {
        printf("[%s][%d] malloc memory fail!!!\n", __func__, __LINE__);
    }
 
    /* create plan for in-place forward DFT */
    plan = kml_fft_mpi_plan_many_dft(rank, n, howmany, block0, block1, in, out, comm,
                                KML_FFT_FORWARD, KML_FFT_MPI_TRANSPOSED_IN);    
         
    /* compute transforms, in-place, as many times as desired */
    kml_fft_execute_dft(plan, in, out);
 
    kml_fft_destroy_plan(plan);
 
    MPI_Finalize();
父主题： C2C变换