kml_fft(f/h)_mpi_local_size_2d

描述了本地进程数据的大小和位置,以及计算需要分配的空间。

接口定义

C interface:

ptrdiff_t kml_fft_mpi_local_size_2d(ptrdiff_t n0, ptrdiff_t n1, MPI_Comm comm, ptrdiff_t *local_n0 ptrdiff_t *local_0_start);

ptrdiff_t kml_fftf_mpi_local_size_2d(ptrdiff_t n0, ptrdiff_t n1, MPI_Comm comm, ptrdiff_t *local_n0, ptrdiff_t *local_0_start);

ptrdiff_t kml_ffth_mpi_local_size_2d(ptrdiff_t n0, ptrdiff_t n1, MPI_Comm comm, ptrdiff_t *local_n0, ptrdiff_t *local_0_start);

返回值

函数返回一个ptrdiff_t类型的值,表示要分配的元素的数量。

参数

参数名

数据类型

描述

输入/输出

n0

  • ptrdiff_t

2D待处理数据第1维度大小,约束:n0 ≥ 1。

输入

n1

  • ptrdiff_t

2D待处理数据第2维度大小,约束:n1 ≥ 1。

输入

comm

  • MPI_Comm

MPI通信器的句柄。

输入

local_n0

  • ptrdiff_t*

本地进程数据大小。

输出

local_0_start

  • ptrdiff_t*

本地进程数据相对全局数据起点偏移量。

输出

依赖

C: "kfft-mpi.h"

示例

kml_fft(f/h)_mpi_plan_dft_2d的相关示例