starpufft.h 2.5 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667
  1. /* StarPU --- Runtime system for heterogeneous multicore architectures.
  2. *
  3. * Copyright (C) 2012 Inria
  4. * Copyright (C) 2010,2012,2014-2015,2017 CNRS
  5. * Copyright (C) 2009,2011,2014 Université de Bordeaux
  6. *
  7. * StarPU is free software; you can redistribute it and/or modify
  8. * it under the terms of the GNU Lesser General Public License as published by
  9. * the Free Software Foundation; either version 2.1 of the License, or (at
  10. * your option) any later version.
  11. *
  12. * StarPU is distributed in the hope that it will be useful, but
  13. * WITHOUT ANY WARRANTY; without even the implied warranty of
  14. * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.
  15. *
  16. * See the GNU Lesser General Public License in COPYING.LGPL for more details.
  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. #define STARPUFFT_FORWARD -1
  27. #define STARPUFFT_INVERSE 1
  28. #define __STARPUFFT(name) starpufft_##name
  29. #define __STARPUFFTF(name) starpufftf_##name
  30. #define __STARPUFFTL(name) starpufftl_##name
  31. #define __STARPUFFT_INTERFACE(starpufft,real) \
  32. typedef real _Complex starpufft(complex); \
  33. \
  34. typedef struct starpufft(plan) *starpufft(plan); \
  35. \
  36. starpufft(plan) starpufft(plan_dft_1d)(int n, int sign, unsigned flags); \
  37. starpufft(plan) starpufft(plan_dft_2d)(int n, int m, int sign, unsigned flags); \
  38. starpufft(plan) starpufft(plan_dft_r2c_1d)(int n, unsigned flags); \
  39. starpufft(plan) starpufft(plan_dft_c2r_1d)(int n, unsigned flags); \
  40. \
  41. void *starpufft(malloc)(size_t n); \
  42. void starpufft(free)(void *p); \
  43. \
  44. int starpufft(execute)(starpufft(plan) p, void *in, void *out); \
  45. struct starpu_task *starpufft(start)(starpufft(plan) p, void *in, void *out); \
  46. \
  47. int starpufft(execute_handle)(starpufft(plan) p, starpu_data_handle_t in, starpu_data_handle_t out); \
  48. struct starpu_task *starpufft(start_handle)(starpufft(plan) p, starpu_data_handle_t in, starpu_data_handle_t out); \
  49. \
  50. void starpufft(cleanup)(starpufft(plan) p); \
  51. void starpufft(destroy_plan)(starpufft(plan) p); \
  52. \
  53. void starpufft(startstats)(void); \
  54. void starpufft(stopstats)(void); \
  55. void starpufft(showstats)(FILE *out);
  56. __STARPUFFT_INTERFACE(__STARPUFFT, double)
  57. __STARPUFFT_INTERFACE(__STARPUFFTF, float)
  58. __STARPUFFT_INTERFACE(__STARPUFFTL, long double)
  59. /* Internal use */
  60. extern int starpufft_last_plan_number;