cannam@95: /*
cannam@95:  * Copyright (c) 2003, 2007-11 Matteo Frigo
cannam@95:  * Copyright (c) 2003, 2007-11 Massachusetts Institute of Technology
cannam@95:  *
cannam@95:  * This program is free software; you can redistribute it and/or modify
cannam@95:  * it under the terms of the GNU General Public License as published by
cannam@95:  * the Free Software Foundation; either version 2 of the License, or
cannam@95:  * (at your option) any later version.
cannam@95:  *
cannam@95:  * This program is distributed in the hope that it will be useful,
cannam@95:  * but WITHOUT ANY WARRANTY; without even the implied warranty of
cannam@95:  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
cannam@95:  * GNU General Public License for more details.
cannam@95:  *
cannam@95:  * You should have received a copy of the GNU General Public License
cannam@95:  * along with this program; if not, write to the Free Software
cannam@95:  * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA  02110-1301  USA
cannam@95:  *
cannam@95:  */
cannam@95: 
cannam@95: #include "bench.h"
cannam@95: 
cannam@95: typedef bench_real R;
cannam@95: typedef bench_complex C;
cannam@95: 
cannam@95: typedef struct dofft_closure_s {
cannam@95:      void (*apply)(struct dofft_closure_s *k,
cannam@95: 		   bench_complex *in, bench_complex *out);
cannam@95:      int recopy_input;
cannam@95: } dofft_closure;
cannam@95: 
cannam@95: double dmax(double x, double y);
cannam@95: 
cannam@95: typedef void (*aconstrain)(C *a, int n);
cannam@95: 
cannam@95: void arand(C *a, int n);
cannam@95: void mkreal(C *A, int n);
cannam@95: void mkhermitian(C *A, int rank, const bench_iodim *dim, int stride);
cannam@95: void mkhermitian1(C *a, int n);
cannam@95: void aadd(C *c, C *a, C *b, int n);
cannam@95: void asub(C *c, C *a, C *b, int n);
cannam@95: void arol(C *b, C *a, int n, int nb, int na);
cannam@95: void aphase_shift(C *b, C *a, int n, int nb, int na, double sign);
cannam@95: void ascale(C *a, C alpha, int n);
cannam@95: double acmp(C *a, C *b, int n, const char *test, double tol);
cannam@95: double mydrand(void);
cannam@95: double impulse(dofft_closure *k,
cannam@95: 	       int n, int vecn, 
cannam@95: 	       C *inA, C *inB, C *inC,
cannam@95: 	       C *outA, C *outB, C *outC,
cannam@95: 	       C *tmp, int rounds, double tol);
cannam@95: double linear(dofft_closure *k, int realp,
cannam@95: 	      int n, C *inA, C *inB, C *inC, C *outA,
cannam@95: 	      C *outB, C *outC, C *tmp, int rounds, double tol);
cannam@95: void preserves_input(dofft_closure *k, aconstrain constrain,
cannam@95:                      int n, C *inA, C *inB, C *outB, int rounds);
cannam@95: 
cannam@95: enum { TIME_SHIFT, FREQ_SHIFT };
cannam@95: double tf_shift(dofft_closure *k, int realp, const bench_tensor *sz,
cannam@95: 		int n, int vecn, double sign,
cannam@95: 		C *inA, C *inB, C *outA, C *outB, C *tmp,
cannam@95: 		int rounds, double tol, int which_shift);
cannam@95: 
cannam@95: typedef struct dotens2_closure_s {
cannam@95:      void (*apply)(struct dotens2_closure_s *k, 
cannam@95: 		   int indx0, int ondx0, int indx1, int ondx1);
cannam@95: } dotens2_closure;
cannam@95: 
cannam@95: void bench_dotens2(const bench_tensor *sz0, 
cannam@95: 		   const bench_tensor *sz1, dotens2_closure *k);
cannam@95: 
cannam@95: void accuracy_test(dofft_closure *k, aconstrain constrain,
cannam@95: 		   int sign, int n, C *a, C *b, int rounds, int impulse_rounds,
cannam@95: 		   double t[6]);
cannam@95: 
cannam@95: void accuracy_dft(bench_problem *p, int rounds, int impulse_rounds,
cannam@95: 		  double t[6]);
cannam@95: void accuracy_rdft2(bench_problem *p, int rounds, int impulse_rounds,
cannam@95: 		    double t[6]);
cannam@95: void accuracy_r2r(bench_problem *p, int rounds, int impulse_rounds,
cannam@95: 		  double t[6]);
cannam@95: 
cannam@95: #if defined(BENCHFFT_LDOUBLE) && HAVE_COSL
cannam@95:    typedef long double trigreal;
cannam@95: #  define COS cosl
cannam@95: #  define SIN sinl
cannam@95: #  define TAN tanl
cannam@95: #  define KTRIG(x) (x##L)
cannam@95: #elif defined(BENCHFFT_QUAD) && HAVE_LIBQUADMATH
cannam@95:    typedef __float128 trigreal;
cannam@95: #  define COS cosq
cannam@95: #  define SIN sinq
cannam@95: #  define TAN tanq
cannam@95: #  define KTRIG(x) (x##Q)
cannam@95: extern trigreal cosq(trigreal);
cannam@95: extern trigreal sinq(trigreal);
cannam@95: extern trigreal tanq(trigreal);
cannam@95: #else
cannam@95:    typedef double trigreal;
cannam@95: #  define COS cos
cannam@95: #  define SIN sin
cannam@95: #  define TAN tan
cannam@95: #  define KTRIG(x) (x)
cannam@95: #endif
cannam@95: #define K2PI KTRIG(6.2831853071795864769252867665590057683943388)