mpi_lu-float.h 1.3 KB

123456789101112131415161718192021222324252627282930313233343536373839404142
  1. /* StarPU --- Runtime system for heterogeneous multicore architectures.
  2. *
  3. * Copyright (C) 2010-2020 Université de Bordeaux, CNRS (LaBRI UMR 5800), Inria
  4. *
  5. * StarPU is free software; you can redistribute it and/or modify
  6. * it under the terms of the GNU Lesser General Public License as published by
  7. * the Free Software Foundation; either version 2.1 of the License, or (at
  8. * your option) any later version.
  9. *
  10. * StarPU is distributed in the hope that it will be useful, but
  11. * WITHOUT ANY WARRANTY; without even the implied warranty of
  12. * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.
  13. *
  14. * See the GNU Lesser General Public License in COPYING.LGPL for more details.
  15. */
  16. #define TYPE float
  17. #define MPI_TYPE MPI_FLOAT
  18. #define STARPU_PLU(name) starpu_pslu_##name
  19. #define CUBLAS_GEMM cublasSgemm
  20. #define CUBLAS_TRSM cublasStrsm
  21. #define CUBLAS_SCAL cublasSscal
  22. #define CUBLAS_GER cublasSger
  23. #define CUBLAS_SWAP cublasSswap
  24. #define CUBLAS_IAMAX cublasIsamax
  25. #define CPU_GEMM STARPU_SGEMM
  26. #define CPU_GEMV STARPU_SGEMV
  27. #define CPU_TRSM STARPU_STRSM
  28. #define CPU_SCAL STARPU_SSCAL
  29. #define CPU_GER STARPU_SGER
  30. #define CPU_SWAP STARPU_SSWAP
  31. #define CPU_TRMM STARPU_STRMM
  32. #define CPU_AXPY STARPU_SAXPY
  33. #define CPU_ASUM STARPU_SASUM
  34. #define CPU_IAMAX STARPU_ISAMAX
  35. #define PIVOT_THRESHHOLD 10e-5