kml_fft(f)_plan_with_nthreads
kml_fft(f)_plan_with_nthreads用于指定FFT库接口函数执行的线程数。(此函数不可重入。)
接口定义
C interface:
void kml_fft_plan_with_nthreads(int nthreads);
void kml_fftf_plan_with_nthreads(int nthreads);
Fortran interface:
CALL KML_FFT_PLAN_WITH_NTHREADS(NTHREADS);
CALL KML_FFTF_PLAN_WITH_NTHREADS(NTHREADS);
参数
参数名 |
数据类型 |
描述 |
输入/输出 |
---|---|---|---|
nthreads |
int |
执行后续FFT变换plan所使用的线程数。 |
输入 |
依赖
C: "kfft.h"
示例
C interface:
kml_fft_init_threads(); int nthreads = 4; kml_fft_plan_with_nthreads(nthreads); int N = 16; kml_fft_complex *in; kml_fft_complex *out; kml_fft_plan plan; in = (kml_fft_complex*)kml_fft_malloc(sizeof(kml_fft_complex) * N); out = (kml_fft_complex*)kml_fft_malloc(sizeof(kml_fft_complex) * N); for (int i = 0; i < N; i++){ in[i][0] = (double)i; in[i][1] = (double)i; } plan = kml_fft_plan_dft_1d(N, in , out, KML_FFT_FORWARD, KML_FFT_ESTIMATE); kml_fft_execute(plan); kml_fft_cleanup_threads(); kml_fft_destroy_plan(plan); kml_fft_free(in); kml_fft_free(out); /* * out = {{1.200000e+02, 1.200000e+02}, {-4.821872e+01, 3.221872e+01}, * {-2.731371e+01, 1.131371e+01}, {-1.997285e+01, 3.972846e+00}, * {-1.600000e+01, 0.000000e+00}, {-1.334543e+01, -2.654571e+00}, * {-1.131371e+01, -4.686292e+00}, {-9.591299e+00, -6.408701e+00}, * {-8.000000e+00, -8.000000e+00}, {-6.408701e+00, -9.591299e+00}, * {-4.686292e+00, -1.131371e+01}, {-2.654571e+00, -1.334543e+01}, * {0.000000e+00, -1.600000e+01}, {3.972846e+00, -1.997285e+01}, * {1.131371e+01, -2.731371e+01}, {3.221872e+01, -4.821872e+01}} */
Fortran interface:
INTEGER(C_INT) :: STATUS STATUS = KML_FFT_INIT_THREADS() INTEGER(C_INT) :: NTHREADS = 4 CALL KML_FFT_PLAN_WITH_NTHREADS(NTHREADS) INTEGER(C_INT) :: N = 16 TYPE(KML_FFT_COMPLEX), POINTER :: IN, OUT TYPE(C_PTR) :: PIN, POUT TYPE(C_PTR) :: PLAN INTEGER(C_SIZE_T) :: SIZE SIZE = 16 * N PIN = KML_FFT_MALLOC(SIZE) POUT = KML_FFT_MALLOC(SIZE) CALL C_F_POINTER(PIN, IN, SHAPE=[N]) CALL C_F_POINTER(POUT, OUT, SHAPE=[N]) INTEGER :: I DO WHILE(I <= N) IN%R = I IN%I = I END DO PLAN = KML_FFT_PLAN_DFT_1D(N, IN , OUT, KML_FFT_FORWARD, KML_FFT_ESTIMATE) CALL KML_FFT_EXECUTE(PLAN, IN, OUT) CALL KML_FFT_DESTROY_PLAN(PLAN) CALL KML_FFT_FREE(PIN) CALL KML_FFT_FREE(POUT) ! ! OUT = \1.200000E+02, 1.200000E+0, -4.821872E+01, 3.221872E+01, ! -2.731371E+01, 1.131371E+01, -1.997285E+01, 3.972846E+00, ! -1.600000E+01, 0.000000E+00, -1.334543E+01, -2.654571E+00, ! -1.131371E+01, -4.686292E+00, -9.591299E+00, -6.408701E+00, ! -8.000000E+00, -8.000000E+00, -6.408701E+00, -9.591299E+00, ! -4.686292E+00, -1.131371E+01, -2.654571E+00, -1.334543E+01, ! 0.000000E+00, -1.600000E+01, 3.972846E+00, -1.997285E+01, ! 1.131371E+01, -2.731371E+01, 3.221872E+01, -4.821872E+01/ !
父主题: 线程函数