diff options
author | Ian Jauslin <ian@jauslin.org> | 2022-05-18 23:52:01 +0200 |
---|---|---|
committer | Ian Jauslin <ian@jauslin.org> | 2022-05-18 23:52:01 +0200 |
commit | 8bce8632c520acea86b4a6b2b7cf3eafe7964124 (patch) | |
tree | 583cb335ce38e2c520c3d84483c4e9b7a23be05a /src/navier-stokes.h | |
parent | 199b8f0df5adeaaac4ca2afc2eff5237dfee36c3 (diff) |
Multithread fft
Diffstat (limited to 'src/navier-stokes.h')
-rw-r--r-- | src/navier-stokes.h | 8 |
1 files changed, 4 insertions, 4 deletions
diff --git a/src/navier-stokes.h b/src/navier-stokes.h index d144fce..707dc52 100644 --- a/src/navier-stokes.h +++ b/src/navier-stokes.h @@ -12,17 +12,17 @@ typedef struct fft_vects { } fft_vect; // compute u_k -int uk( int K1, int K2, int N1, int N2, unsigned int nsteps, double nu, double delta, _Complex double (*g)(int,int), unsigned int print_freq); +int uk( int K1, int K2, int N1, int N2, unsigned int nsteps, double nu, double delta, _Complex double (*g)(int,int), unsigned int print_freq, unsigned int nthreads); // compute enstrophy -int enstrophy( int K1, int K2, int N1, int N2, unsigned int nsteps, double nu, double delta, _Complex double (*g)(int,int), unsigned int print_freq); +int enstrophy( int K1, int K2, int N1, int N2, unsigned int nsteps, double nu, double delta, _Complex double (*g)(int,int), unsigned int print_freq, unsigned int nthreads); // compute solution as a function of time, but do not print anything (useful for debugging) -int quiet( int K1, int K2, int N1, int N2, unsigned int nsteps, double nu, double delta, _Complex double (*g)(int,int)); +int quiet( int K1, int K2, int N1, int N2, unsigned int nsteps, double nu, double delta, _Complex double (*g)(int,int), unsigned int nthreads); // initialize vectors for computation -int ns_init_tmps( _Complex double **u, _Complex double ** tmp1, _Complex double **tmp2, _Complex double **tmp3, fft_vect* fft1, fft_vect *fft2, fft_vect *ifft, int K1, int K2, int N1, int N2); +int ns_init_tmps( _Complex double **u, _Complex double ** tmp1, _Complex double **tmp2, _Complex double **tmp3, fft_vect* fft1, fft_vect *fft2, fft_vect *ifft, int K1, int K2, int N1, int N2, unsigned int nthreads); // release vectors int ns_free_tmps( _Complex double* u, _Complex double* tmp1, _Complex double *tmp2,_Complex double *tmp3, fft_vect fft1, fft_vect fft2, fft_vect ifft); |