1 #ifndef TEST_MAIN_H 2 #define TEST_MAIN_H 3 4 /* max # of dimensions accepted on the command line */ 5 #define MAX_CMDLINE_RANK 20 6 7 struct size { 8 int rank; 9 int is_nd; 10 int narray[MAX_CMDLINE_RANK]; 11 }; 12 13 /******************** 14 * macrology: 15 ********************/ 16 #ifndef TRUE 17 #define TRUE 42 18 #endif 19 #ifndef FALSE 20 #define FALSE (!TRUE) 21 #endif 22 23 #define CHECK(condition, complaint) \ 24 if (!(condition)) { \ 25 fflush(stdout); \ 26 fprintf(stderr, "FATAL ERROR: %s\n", complaint); \ 27 fftw_die("failed test.\n"); \ 28 } 29 30 #define WHEN_VERBOSE(a, x) if (verbose >= a) x 31 32 #ifdef FFTW_ENABLE_FLOAT 33 #define TOLERANCE (1e-2) 34 #else 35 #define TOLERANCE (1e-6) 36 #endif 37 38 #define DRAND() mydrand() 39 40 #define SPECIFICP(x) (x ? "specific" : "generic") 41 42 /******************* 43 * global variables 44 *******************/ 45 extern int verbose; 46 extern int only_one_speed_test; 47 extern int speed_flag, wisdom_flag, measure_flag, no_vector_flag; 48 extern int chk_mem_leak; 49 extern int paranoid; 50 extern int howmany_fields; 51 extern int io_okay; 52 53 /* Time an FFT routine, invoked by fft. a is the array being 54 * transformed, n is its total length. t should be a variable 55 * --the time (sec) per fft is assigned to it. */ 56 57 #define FFTW_TIME_FFT(fft,a,n,t) \ 58 { \ 59 fftw_time ts,te; \ 60 double total_t; \ 61 int tfft_iters = 1, tfft_iter; \ 62 zero_arr((n), (a)); \ 63 do { \ 64 ts = fftw_get_time(); \ 65 for (tfft_iter = 0; tfft_iter < tfft_iters; ++tfft_iter) fft; \ 66 te = fftw_get_time(); \ 67 t = (total_t=fftw_time_to_sec(fftw_time_diff(te,ts))) / tfft_iters; \ 68 tfft_iters *= 2; \ 69 } while (total_t < 2.0); \ 70 } 71 72 #define MAX_STRIDE 3 73 #define MAX_HOWMANY 3 74 #define MAX_RANK 5 75 #define PLANNER_TEST_SIZE 100 76 77 extern int coinflip(void); 78 extern double mydrand(void); 79 extern char *smart_sprint_time(double x); 80 extern void please_wait(void); 81 extern void please_wait_forever(void); 82 extern double mflops(double t, int N); 83 extern void print_dims(struct size sz); 84 85 #define SQR(x) ((x) * (x)) 86 87 extern double compute_error_complex(fftw_complex * A, int astride, 88 fftw_complex * B, int bstride, int n); 89 90 extern fftw_direction random_dir(void); 91 92 /*** the following symbols should be defined in fftw_test.c/rfftw_test.c ***/ 93 extern char fftw_prefix[]; 94 extern void test_speed_aux(int n, fftw_direction dir, int flags, int specific); 95 extern void test_speed_nd_aux(struct size sz, fftw_direction dir, 96 int flags, int specific); 97 extern void test_correctness(int n); 98 extern void testnd_correctness(struct size sz, fftw_direction dir, 99 int alt_api, int specific, int force_buf); 100 extern void test_planner(int rank); 101 102 extern void test_init(int *argc, char ***argv); 103 extern void test_finish(void); 104 extern void enter_paranoid_mode(void); 105 106 extern int get_option(int argc, char **argv, 107 char *argval, int argval_maxlen); 108 extern int default_get_option(int argc, char **argv, 109 char *argval, int argval_maxlen); 110 111 #endif /* TEST_MAIN_H */ 112