complex_double.h 1.6 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152
  1. /* StarPU --- Runtime system for heterogeneous multicore architectures.
  2. *
  3. * Copyright (C) 2011,2012 Inria
  4. * Copyright (C) 2010,2012,2015,2017 CNRS
  5. * Copyright (C) 2009,2010,2014,2015,2017 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. /* Complex double macros */
  19. #include <complex.h>
  20. #include "blas_complex.h"
  21. #define TYPE complex double
  22. #define CUBLAS_TYPE cuDoubleComplex
  23. #define STARPU_LU(name) starpu_zlu_##name
  24. #define COMPLEX_LU
  25. #define CUBLAS_GEMM cublasZgemm
  26. #define CUBLAS_TRSM cublasZtrsm
  27. #define CUBLAS_SCAL cublasZscal
  28. #define CUBLAS_GER cublasZgeru
  29. #define CUBLAS_SWAP cublasZswap
  30. #define CUBLAS_IAMAX cublasIzamax
  31. #define CPU_GEMM ZGEMM
  32. #define CPU_TRSM ZTRSM
  33. #define CPU_SCAL ZSCAL
  34. #define CPU_GER ZGERU
  35. #define CPU_SWAP ZSWAP
  36. #define CPU_TRMM ZTRMM
  37. #define CPU_AXPY ZAXPY
  38. #define CPU_ASUM DZASUM
  39. #define CPU_IAMAX IZAMAX
  40. #define PIVOT_THRESHHOLD 10e-5
  41. #define CAN_EXECUTE .can_execute = can_execute,
  42. #define ISZERO(f) (fpclassify(creal(f)) == FP_ZERO && fpclassify(cimag(f)) == FP_ZERO)