complex_float.h 1.3 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849
  1. /* StarPU --- Runtime system for heterogeneous multicore architectures.
  2. *
  3. * Copyright (C) 2009, 2010, 2014-2015 Université de Bordeaux
  4. * Copyright (C) 2010 CNRS
  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. /* Complex float macros */
  18. #include <complex.h>
  19. #include "blas_complex.h"
  20. #define TYPE complex float
  21. #define CUBLAS_TYPE cuComplex
  22. #define STARPU_LU(name) starpu_clu_##name
  23. #define COMPLEX_LU
  24. #define CUBLAS_GEMM cublasCgemm
  25. #define CUBLAS_TRSM cublasCtrsm
  26. #define CUBLAS_SCAL cublasCscal
  27. #define CUBLAS_GER cublasCgeru
  28. #define CUBLAS_SWAP cublasCswap
  29. #define CUBLAS_IAMAX cublasIcamax
  30. #define CPU_GEMM CGEMM
  31. #define CPU_TRSM CTRSM
  32. #define CPU_SCAL CSCAL
  33. #define CPU_GER CGERU
  34. #define CPU_SWAP CSWAP
  35. #define CPU_TRMM CTRMM
  36. #define CPU_AXPY CAXPY
  37. #define CPU_ASUM SCASUM
  38. #define CPU_IAMAX ICAMAX
  39. #define PIVOT_THRESHHOLD 10e-5
  40. #define CAN_EXECUTE