Amesos2_cuSOLVER_TypeMap.hpp

// @HEADER
// *****************************************************************************
//           Amesos2: Templated Direct Sparse Solver Package
//
// Copyright 2011 NTESS and the Amesos2 contributors.
// SPDX-License-Identifier: BSD-3-Clause
// *****************************************************************************
// @HEADER

#ifndef AMESOS2_CUSOLVER_TYPEMAP_HPP
#define AMESOS2_CUSOLVER_TYPEMAP_HPP

#include <functional>
#ifdef HAVE_TEUCHOS_COMPLEX
#include <complex>
#endif

#include "Amesos2_TypeMap.hpp"

namespace Amesos2 {

  // forward declaration due to circular reference
  template <class, class> class cuSOLVER;

  template <>
  struct TypeMap<cuSOLVER,float>
  {
    typedef float type;
    typedef float magnitude_type;
  };

  template <>
  struct TypeMap<cuSOLVER,double>
  {
    typedef double type;
    typedef double magnitude_type;
  };

#ifdef HAVE_TEUCHOS_COMPLEX

  template <>
  struct TypeMap<cuSOLVER,std::complex<float> >
  {
    typedef Kokkos::complex<float> type;
    typedef float magnitude_type;
  };

  template <>
  struct TypeMap<cuSOLVER,Kokkos::complex<float> >
  {
    typedef Kokkos::complex<float> type;
    typedef float magnitude_type;
  };

  template <>
  struct TypeMap<cuSOLVER,std::complex<double> >
  {
    typedef Kokkos::complex<double> type;
    typedef double magnitude_type;
  };

  template <>
  struct TypeMap<cuSOLVER,Kokkos::complex<double> >
  {
    typedef Kokkos::complex<double> type;
    typedef double magnitude_type;
  };

#endif  // HAVE_TEUCHOS_COMPLEX

} // end namespace Amesos

#endif  // AMESOS2_CUSOLVER_TYPEMAP_HPP