kml_fft(f/h)_mpi_local_size
描述了本地进程数据的大小和位置,以及计算需要分配的空间。
接口定义
C interface:
ptrdiff_t kml_fft_mpi_local_size (int rank, const ptrdiff_t *n, MPI_Comm comm, ptrdiff_t *local_n0 ptrdiff_t *local_0_start);
ptrdiff_t kml_fftf_mpi_ local_size (int rank, const ptrdiff_t *n, MPI_Comm comm, ptrdiff_t *local_n0, ptrdiff_t *local_0_start);
ptrdiff_t kml_ffth_mpi_ local_size (int rank, const ptrdiff_t *n, MPI_Comm comm, ptrdiff_t *local_n0, ptrdiff_t *local_0_start);
返回值
函数返回一个ptrdiff_t类型的值,表示要分配的元素的数量。
参数
参数名 |
数据类型 |
描述 |
输入/输出 |
---|---|---|---|
rank |
|
待处理数据维度数量,约束:rank ≥ 1。 |
输入 |
n |
|
n是维度为rank的数组,包含数据序列每一维度的大小,约束:n[i] ≥ 1, for i in 0 to rank - 1。 |
输入 |
comm |
|
MPI通信器的句柄。 |
输入 |
local_n0 |
|
本地进程数据大小。 |
输出 |
local_0_start |
|
本地进程数据相对全局数据起点偏移量。 |
输出 |
依赖
C: "kfft-mpi.h"
示例
父主题: MPI Local Size函数