Chris@10
|
1 /*
|
Chris@10
|
2 * Copyright (c) 2003, 2007-11 Matteo Frigo
|
Chris@10
|
3 * Copyright (c) 2003, 2007-11 Massachusetts Institute of Technology
|
Chris@10
|
4 *
|
Chris@10
|
5 * This program is free software; you can redistribute it and/or modify
|
Chris@10
|
6 * it under the terms of the GNU General Public License as published by
|
Chris@10
|
7 * the Free Software Foundation; either version 2 of the License, or
|
Chris@10
|
8 * (at your option) any later version.
|
Chris@10
|
9 *
|
Chris@10
|
10 * This program is distributed in the hope that it will be useful,
|
Chris@10
|
11 * but WITHOUT ANY WARRANTY; without even the implied warranty of
|
Chris@10
|
12 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
Chris@10
|
13 * GNU General Public License for more details.
|
Chris@10
|
14 *
|
Chris@10
|
15 * You should have received a copy of the GNU General Public License
|
Chris@10
|
16 * along with this program; if not, write to the Free Software
|
Chris@10
|
17 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
|
Chris@10
|
18 *
|
Chris@10
|
19 */
|
Chris@10
|
20
|
Chris@10
|
21
|
Chris@10
|
22 /*
|
Chris@10
|
23 * This header file must include every file or define every
|
Chris@10
|
24 * type or macro which is required to compile a codelet.
|
Chris@10
|
25 */
|
Chris@10
|
26
|
Chris@10
|
27 #ifndef __DFT_CODELET_H__
|
Chris@10
|
28 #define __DFT_CODELET_H__
|
Chris@10
|
29
|
Chris@10
|
30 #include "ifftw.h"
|
Chris@10
|
31
|
Chris@10
|
32 /**************************************************************
|
Chris@10
|
33 * types of codelets
|
Chris@10
|
34 **************************************************************/
|
Chris@10
|
35
|
Chris@10
|
36 /* DFT codelets */
|
Chris@10
|
37 typedef struct kdft_desc_s kdft_desc;
|
Chris@10
|
38
|
Chris@10
|
39 typedef struct {
|
Chris@10
|
40 int (*okp)(
|
Chris@10
|
41 const kdft_desc *desc,
|
Chris@10
|
42 const R *ri, const R *ii, const R *ro, const R *io,
|
Chris@10
|
43 INT is, INT os, INT vl, INT ivs, INT ovs,
|
Chris@10
|
44 const planner *plnr);
|
Chris@10
|
45 INT vl;
|
Chris@10
|
46 } kdft_genus;
|
Chris@10
|
47
|
Chris@10
|
48 struct kdft_desc_s {
|
Chris@10
|
49 INT sz; /* size of transform computed */
|
Chris@10
|
50 const char *nam;
|
Chris@10
|
51 opcnt ops;
|
Chris@10
|
52 const kdft_genus *genus;
|
Chris@10
|
53 INT is;
|
Chris@10
|
54 INT os;
|
Chris@10
|
55 INT ivs;
|
Chris@10
|
56 INT ovs;
|
Chris@10
|
57 };
|
Chris@10
|
58
|
Chris@10
|
59 typedef void (*kdft) (const R *ri, const R *ii, R *ro, R *io,
|
Chris@10
|
60 stride is, stride os, INT vl, INT ivs, INT ovs);
|
Chris@10
|
61 void X(kdft_register)(planner *p, kdft codelet, const kdft_desc *desc);
|
Chris@10
|
62
|
Chris@10
|
63
|
Chris@10
|
64 typedef struct ct_desc_s ct_desc;
|
Chris@10
|
65
|
Chris@10
|
66 typedef struct {
|
Chris@10
|
67 int (*okp)(
|
Chris@10
|
68 const struct ct_desc_s *desc,
|
Chris@10
|
69 const R *rio, const R *iio,
|
Chris@10
|
70 INT rs, INT vs, INT m, INT mb, INT me, INT ms,
|
Chris@10
|
71 const planner *plnr);
|
Chris@10
|
72 INT vl;
|
Chris@10
|
73 } ct_genus;
|
Chris@10
|
74
|
Chris@10
|
75 struct ct_desc_s {
|
Chris@10
|
76 INT radix;
|
Chris@10
|
77 const char *nam;
|
Chris@10
|
78 const tw_instr *tw;
|
Chris@10
|
79 const ct_genus *genus;
|
Chris@10
|
80 opcnt ops;
|
Chris@10
|
81 INT rs;
|
Chris@10
|
82 INT vs;
|
Chris@10
|
83 INT ms;
|
Chris@10
|
84 };
|
Chris@10
|
85
|
Chris@10
|
86 typedef void (*kdftw) (R *rioarray, R *iioarray, const R *W,
|
Chris@10
|
87 stride ios, INT mb, INT me, INT ms);
|
Chris@10
|
88 void X(kdft_dit_register)(planner *p, kdftw codelet, const ct_desc *desc);
|
Chris@10
|
89 void X(kdft_dif_register)(planner *p, kdftw codelet, const ct_desc *desc);
|
Chris@10
|
90
|
Chris@10
|
91
|
Chris@10
|
92 typedef void (*kdftwsq) (R *rioarray, R *iioarray,
|
Chris@10
|
93 const R *W, stride is, stride vs,
|
Chris@10
|
94 INT mb, INT me, INT ms);
|
Chris@10
|
95 void X(kdft_difsq_register)(planner *p, kdftwsq codelet, const ct_desc *desc);
|
Chris@10
|
96
|
Chris@10
|
97
|
Chris@10
|
98 extern const solvtab X(solvtab_dft_standard);
|
Chris@10
|
99 extern const solvtab X(solvtab_dft_sse2);
|
Chris@10
|
100 extern const solvtab X(solvtab_dft_avx);
|
Chris@10
|
101 extern const solvtab X(solvtab_dft_altivec);
|
Chris@10
|
102 extern const solvtab X(solvtab_dft_neon);
|
Chris@10
|
103
|
Chris@10
|
104 #endif /* __DFT_CODELET_H__ */
|