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