annotate src/fftw-3.3.8/support/addchain.c @ 84:08ae793730bd

Add null config files
author Chris Cannam
date Mon, 02 Mar 2020 14:03:47 +0000
parents d0c2a83c1364
children
rev   line source
Chris@82 1 /* addition-chain optimizer */
Chris@82 2 #include <stdio.h>
Chris@82 3 #include <stdlib.h>
Chris@82 4 #include <unistd.h>
Chris@82 5
Chris@82 6 static int verbose;
Chris@82 7 static int mulcost = 18;
Chris@82 8 static int ldcost = 2;
Chris@82 9 static int sqcost = 10;
Chris@82 10 static int reflcost = 8;
Chris@82 11 #define INFTY 100000
Chris@82 12
Chris@82 13 static int *answer;
Chris@82 14 static int best_so_far;
Chris@82 15
Chris@82 16 static void print_answer(int n, int t)
Chris@82 17 {
Chris@82 18 int i;
Chris@82 19 printf("| (%d, %d) -> [", n, t);
Chris@82 20 for (i = 0; i < t; ++i)
Chris@82 21 printf("%d;", answer[i]);
Chris@82 22 printf("] (* %d *)\n", best_so_far);
Chris@82 23 }
Chris@82 24
Chris@82 25 #define DO(i, j, k, cst) \
Chris@82 26 if (k < n) { \
Chris@82 27 int c = A[i] + A[j] + cst; \
Chris@82 28 if (c < A[k]) { \
Chris@82 29 A[k] = c; \
Chris@82 30 changed = 1; \
Chris@82 31 } \
Chris@82 32 }
Chris@82 33
Chris@82 34 #define DO3(i, j, l, k, cst) \
Chris@82 35 if (k < n) { \
Chris@82 36 int c = A[i] + A[j] + A[l] + cst; \
Chris@82 37 if (c < A[k]) { \
Chris@82 38 A[k] = c; \
Chris@82 39 changed = 1; \
Chris@82 40 } \
Chris@82 41 }
Chris@82 42
Chris@82 43 static int optimize(int n, int *A)
Chris@82 44 {
Chris@82 45 int i, j, k, changed, cst, cstmax;
Chris@82 46
Chris@82 47 do {
Chris@82 48 changed = 0;
Chris@82 49 for (i = 0; i < n; ++i) {
Chris@82 50 k = i + i;
Chris@82 51 DO(i, i, k, sqcost);
Chris@82 52 }
Chris@82 53
Chris@82 54 for (i = 0; i < n; ++i) {
Chris@82 55 for (j = 0; j <= i; ++j) {
Chris@82 56 k = i + j;
Chris@82 57 DO(i, j, k, mulcost);
Chris@82 58 k = i - j;
Chris@82 59 DO(i, j, k, mulcost);
Chris@82 60
Chris@82 61 k = i + j;
Chris@82 62 DO3(i, j, i - j, k, reflcost);
Chris@82 63 }
Chris@82 64 }
Chris@82 65
Chris@82 66 } while (changed);
Chris@82 67
Chris@82 68 cst = cstmax = 0;
Chris@82 69 for (i = 0; i < n; ++i) {
Chris@82 70 cst += A[i];
Chris@82 71 if (A[i] > cstmax) cstmax = A[i];
Chris@82 72 }
Chris@82 73 /* return cstmax; */
Chris@82 74 return cst;
Chris@82 75 }
Chris@82 76
Chris@82 77 static void search(int n, int t, int *A, int *B, int depth)
Chris@82 78 {
Chris@82 79 if (depth == 0) {
Chris@82 80 int i, tc;
Chris@82 81 for (i = 0; i < n; ++i)
Chris@82 82 A[i] = INFTY;
Chris@82 83 A[0] = 0; /* always free */
Chris@82 84 for (i = 1; i <= t; ++i)
Chris@82 85 A[B[-i]] = ldcost;
Chris@82 86
Chris@82 87 tc = optimize(n, A);
Chris@82 88 if (tc < best_so_far) {
Chris@82 89 best_so_far = tc;
Chris@82 90 for (i = 1; i <= t; ++i)
Chris@82 91 answer[t - i] = B[-i];
Chris@82 92 if (verbose)
Chris@82 93 print_answer(n, t);
Chris@82 94 }
Chris@82 95 } else {
Chris@82 96 for (B[0] = B[-1] + 1; B[0] < n; ++B[0])
Chris@82 97 search(n, t, A, B + 1, depth - 1);
Chris@82 98 }
Chris@82 99 }
Chris@82 100
Chris@82 101 static void doit(int n, int t)
Chris@82 102 {
Chris@82 103 int *A;
Chris@82 104 int *B;
Chris@82 105
Chris@82 106 A = malloc(n * sizeof(int));
Chris@82 107 B = malloc((t + 1) * sizeof(int));
Chris@82 108 answer = malloc(t * sizeof(int));
Chris@82 109
Chris@82 110 B[0] = 0;
Chris@82 111 best_so_far = INFTY;
Chris@82 112 search(n, t, A, B + 1, t);
Chris@82 113
Chris@82 114 print_answer(n, t);
Chris@82 115
Chris@82 116 free(A); free(B); free(answer);
Chris@82 117 }
Chris@82 118
Chris@82 119 int main(int argc, char *argv[])
Chris@82 120 {
Chris@82 121 int n = 32;
Chris@82 122 int t = 3;
Chris@82 123 int all;
Chris@82 124 int ch;
Chris@82 125
Chris@82 126 verbose = 0;
Chris@82 127 all = 0;
Chris@82 128 while ((ch = getopt(argc, argv, "n:t:m:l:r:s:va")) != -1) {
Chris@82 129 switch (ch) {
Chris@82 130 case 'n':
Chris@82 131 n = atoi(optarg);
Chris@82 132 break;
Chris@82 133 case 't':
Chris@82 134 t = atoi(optarg);
Chris@82 135 break;
Chris@82 136 case 'm':
Chris@82 137 mulcost = atoi(optarg);
Chris@82 138 break;
Chris@82 139 case 'l':
Chris@82 140 ldcost = atoi(optarg);
Chris@82 141 break;
Chris@82 142 case 's':
Chris@82 143 sqcost = atoi(optarg);
Chris@82 144 break;
Chris@82 145 case 'r':
Chris@82 146 reflcost = atoi(optarg);
Chris@82 147 break;
Chris@82 148 case 'v':
Chris@82 149 ++verbose;
Chris@82 150 break;
Chris@82 151 case 'a':
Chris@82 152 ++all;
Chris@82 153 break;
Chris@82 154 case '?':
Chris@82 155 fprintf(stderr, "use the source\n");
Chris@82 156 exit(1);
Chris@82 157 }
Chris@82 158 }
Chris@82 159
Chris@82 160 if (all) {
Chris@82 161 for (n = 4; n <= 64; n *= 2) {
Chris@82 162 int n1 = n - 1; if (n1 > 7) n1 = 7;
Chris@82 163 for (t = 1; t <= n1; ++t)
Chris@82 164 doit(n, t);
Chris@82 165 }
Chris@82 166 } else {
Chris@82 167 doit(n, t);
Chris@82 168 }
Chris@82 169
Chris@82 170 return 0;
Chris@82 171 }