Actual source code: slepcmagma.h

slepc-3.15.0 2021-03-31
Report Typos and Errors
  1: /*
  2:    - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -
  3:    SLEPc - Scalable Library for Eigenvalue Problem Computations
  4:    Copyright (c) 2002-2021, Universitat Politecnica de Valencia, Spain

  6:    This file is part of SLEPc.
  7:    SLEPc is distributed under a 2-clause BSD license (see LICENSE).
  8:    - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -
  9: */
 10: /*
 11:    Macro definitions to use MAGMA functionality
 12: */

 14: #if !defined(SLEPCMAGMA_H)
 15: #define SLEPCMAGMA_H

 17: #include <magma_v2.h>
 18: #define CHKERRMAGMA(mierr) CHKERRABORT(PETSC_COMM_SELF,mierr)

 20: #if defined(PETSC_USE_COMPLEX)
 21: #if defined(PETSC_USE_REAL_SINGLE)
 22: #define magma_xgeev(a,b,c,d,e,f,g,h,i,j,k,l.m.n) magma_cgeev((a),(b),(c),(magmaFloatComplex*)(d),(e),(magmaFloatComplex*)(f),(magmaFloatComplex*)(g),(h),(magmaFloatComplex*)(i),(j),(magmaFloatComplex*)(k),(l),(m),(n))
 23: #define magma_xgesv_gpu(a,b,c,d,e,f,g,h)         magma_cgesv_gpu((a),(b),(magmaFloatComplex_ptr)(c),(d),(e),(magmaFloatComplex_ptr)(f),(g),(h))
 24: #define magma_xgetrf_gpu(a,b,c,d,e,f)   magma_cgetrf_gpu((a),(b),(magmaFloatComplex_ptr)(c),(d),(e),(f))
 25: #define magma_xgetri_gpu(a,b,c,d,e,f,g) magma_cgetri_gpu((a),(magmaFloatComplex_ptr)(b),(c),(d),(magmaFloatComplex_ptr)(e),(f),(g))
 26: #define magma_get_xgetri_nb             magma_get_cgetri_nb
 27: #else
 28: #define magma_xgeev(a,b,c,d,e,f,g,h,i,j,k,l,m,n) magma_zgeev((a),(b),(c),(magmaDoubleComplex*)(d),(e),(magmaDoubleComplex*)(f),(magmaDoubleComplex*)(g),(h),(magmaDoubleComplex*)(i),(j),(magmaDoubleComplex*)(k),(l),(m),(n))
 29: #define magma_xgesv_gpu(a,b,c,d,e,f,g,h)         magma_zgesv_gpu((a),(b),(magmaDoubleComplex_ptr)(c),(d),(e),(magmaDoubleComplex_ptr)(f),(g),(h))
 30: #define magma_xgetrf_gpu(a,b,c,d,e,f)   magma_zgetrf_gpu((a),(b),(magmaDoubleComplex_ptr)(c),(d),(e),(f))
 31: #define magma_xgetri_gpu(a,b,c,d,e,f,g) magma_zgetri_gpu((a),(magmaDoubleComplex_ptr)(b),(c),(d),(magmaDoubleComplex_ptr)(e),(f),(g))
 32: #define magma_get_xgetri_nb             magma_get_zgetri_nb
 33: #endif
 34: #else
 35: #if defined(PETSC_USE_REAL_SINGLE)
 36: #define magma_xgeev                     magma_sgeev
 37: #define magma_xgesv_gpu                 magma_sgesv_gpu
 38: #define magma_xgetrf_gpu                magma_sgetrf_gpu
 39: #define magma_xgetri_gpu                magma_sgetri_gpu
 40: #define magma_get_xgetri_nb             magma_get_sgetri_nb
 41: #else
 42: #define magma_xgeev                     magma_dgeev
 43: #define magma_xgesv_gpu                 magma_dgesv_gpu
 44: #define magma_xgetrf_gpu                magma_dgetrf_gpu
 45: #define magma_xgetri_gpu                magma_dgetri_gpu
 46: #define magma_get_xgetri_nb             magma_get_dgetri_nb
 47: #endif
 48: #endif

 50: #if defined(PETSC_USE_REAL_SINGLE)
 51: #define magma_Cgesv_gpu(a,b,c,d,e,f,g,h)         magma_cgesv_gpu((a),(b),(magmaFloatComplex_ptr)(c),(d),(e),(magmaFloatComplex_ptr)(f),(g),(h))
 52: #else
 53: #define magma_Cgesv_gpu(a,b,c,d,e,f,g,h)         magma_zgesv_gpu((a),(b),(magmaDoubleComplex_ptr)(c),(d),(e),(magmaDoubleComplex_ptr)(f),(g),(h))
 54: #endif

 56: #endif