StarPU Handbook
 All Data Structures Files Functions Variables Typedefs Enumerations Enumerator Groups Pages
starpufft.h
1 /* StarPU --- Runtime system for heterogeneous multicore architectures.
2  *
3  * Copyright (C) 2009, 2011 Université de Bordeaux 1
4  * Copyright (C) 2010, 2012 Centre National de la Recherche Scientifique
5  *
6  * StarPU is free software; you can redistribute it and/or modify
7  * it under the terms of the GNU Lesser General Public License as published by
8  * the Free Software Foundation; either version 2.1 of the License, or (at
9  * your option) any later version.
10  *
11  * StarPU is distributed in the hope that it will be useful, but
12  * WITHOUT ANY WARRANTY; without even the implied warranty of
13  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.
14  *
15  * See the GNU Lesser General Public License in COPYING.LGPL for more details.
16  */
17 
18 #include <stdio.h>
19 #include <complex.h>
20 #include <starpu.h>
21 #ifdef STARPU_USE_CUDA
22 #include <cufft.h>
23 #define STARPU_CUFFT_REPORT_ERROR(status) \
24  STARPUFFT(report_error)(__starpu_func__, __FILE__, __LINE__, status)
25 #endif /* !STARPU_USE_CUDA */
26 
27 #define STARPUFFT_FORWARD -1
28 #define STARPUFFT_INVERSE 1
29 
30 #define __STARPUFFT(name) starpufft_##name
31 #define __STARPUFFTF(name) starpufftf_##name
32 #define __STARPUFFTL(name) starpufftl_##name
33 
34 #define __STARPUFFT_INTERFACE(starpufft,real) \
35 typedef real _Complex starpufft(complex); \
36 \
37 typedef struct starpufft(plan) *starpufft(plan); \
38 \
39 starpufft(plan) starpufft(plan_dft_1d)(int n, int sign, unsigned flags); \
40 starpufft(plan) starpufft(plan_dft_2d)(int n, int m, int sign, unsigned flags); \
41 starpufft(plan) starpufft(plan_dft_r2c_1d)(int n, unsigned flags); \
42 starpufft(plan) starpufft(plan_dft_c2r_1d)(int n, unsigned flags); \
43 \
44 void *starpufft(malloc)(size_t n); \
45 void starpufft(free)(void *p); \
46 \
47 int starpufft(execute)(starpufft(plan) p, void *in, void *out); \
48 struct starpu_task *starpufft(start)(starpufft(plan) p, void *in, void *out); \
49 \
50 int starpufft(execute_handle)(starpufft(plan) p, starpu_data_handle_t in, starpu_data_handle_t out); \
51 struct starpu_task *starpufft(start_handle)(starpufft(plan) p, starpu_data_handle_t in, starpu_data_handle_t out); \
52 \
53 void starpufft(cleanup)(starpufft(plan) p); \
54 void starpufft(destroy_plan)(starpufft(plan) p); \
55 \
56 void starpufft(startstats)(void); \
57 void starpufft(stopstats)(void); \
58 void starpufft(showstats)(FILE *out);
59 
60 __STARPUFFT_INTERFACE(__STARPUFFT, double)
61 __STARPUFFT_INTERFACE(__STARPUFFTF, float)
62 __STARPUFFT_INTERFACE(__STARPUFFTL, long double)
63 
64 /* Internal use */
65 extern int starpufft_last_plan_number;