Amesos2_Superlu_TypeMap.hpp

Go to the documentation of this file.

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

#ifndef AMESOS2_SUPERLU_TYPEMAP_HPP
#define AMESOS2_SUPERLU_TYPEMAP_HPP

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

#include <Teuchos_as.hpp>
#ifdef HAVE_TEUCHOS_COMPLEX
#include <Teuchos_SerializationTraits.hpp>
#endif

#include "Amesos2_TypeMap.hpp"

/* The SuperLU comples headers file only need to be included if
   complex has been enabled in Teuchos.  In addition we only need to
   define the conversion and printing functions if complex has been
   enabled. */
namespace SLU {

typedef int int_t;

extern "C" {

#undef __SUPERLU_SUPERMATRIX
#include "supermatrix.h"  // for Dtype_t declaration

#ifdef HAVE_TEUCHOS_COMPLEX
namespace C {
#undef __SUPERLU_SCOMPLEX
#undef SCOMPLEX_INCLUDE
#include "slu_scomplex.h"     // single-precision complex data type definitions
}

namespace Z {
#undef __SUPERLU_DCOMPLEX
#undef DCOMPLEX_INCLUDE
#include "slu_dcomplex.h"     // double-precision complex data type definitions
}
#endif // HAVE_TEUCHOS_COMPLEX

} // end extern "C"

} // end namespace SLU

#ifdef HAVE_TEUCHOS_COMPLEX

/* ==================== Conversion ==================== */
namespace Teuchos {

template <>
class ValueTypeConversionTraits<SLU::C::complex, Kokkos::complex<float>>
{
public:
  static SLU::C::complex convert( const Kokkos::complex<float> t ) {
    SLU::C::complex ret;
    ret.r = t.real();
    ret.i = t.imag();
    return( ret );
  }

  static SLU::C::complex safeConvert( const Kokkos::complex<float> t ) {
    SLU::C::complex ret;
    ret.r = t.real();
    ret.i = t.imag();
    return( ret );
  }
};

template <>
class ValueTypeConversionTraits<SLU::Z::doublecomplex, Kokkos::complex<double>>
{
public:
  static SLU::Z::doublecomplex convert( const Kokkos::complex<double> t ) {
    SLU::Z::doublecomplex ret;
    ret.r = t.real();
    ret.i = t.imag();
    return( ret );
  }

  static SLU::Z::doublecomplex safeConvert( const Kokkos::complex<double> t ) {
    SLU::Z::doublecomplex ret;
    ret.r = t.real();
    ret.i = t.imag();
    return( ret );
  }
};

// Also convert from SLU types

template <>
class ValueTypeConversionTraits<Kokkos::complex<float>, SLU::C::complex>
{
public:
  static Kokkos::complex<float> convert( const SLU::C::complex t ) {
    return ( Kokkos::complex<float>( t.r, t.i ) );
  }

  static Kokkos::complex<float> safeConvert( const SLU::C::complex t ) {
    return ( Kokkos::complex<float>( t.r, t.i ) );
  }
};

template <>
class ValueTypeConversionTraits<Kokkos::complex<double>, SLU::Z::doublecomplex>
{
public:
  static Kokkos::complex<double> convert( const SLU::Z::doublecomplex t ) {
    return ( Kokkos::complex<double>( t.r, t.i ) );
  }

  static Kokkos::complex<double> safeConvert( const SLU::Z::doublecomplex t ) {
    return ( Kokkos::complex<double>( t.r, t.i ) );
  }
};


} // end namespace Teuchos

#endif  // HAVE_TEUCHOS_COMPLEX

namespace Amesos2 {

template <class, class> class Superlu;

/* Specialize the Amesos2::TypeMap struct for Superlu types
 *
 * \cond Superlu_type_specializations
 */
template <>
struct TypeMap<Superlu,float>
{
  static SLU::Dtype_t dtype;
  typedef float convert_type;
  typedef float type;
  typedef float magnitude_type;
};


template <>
struct TypeMap<Superlu,double>
{
  static SLU::Dtype_t dtype;
  typedef double convert_type;
  typedef double type;
  typedef double magnitude_type;
};


#ifdef HAVE_TEUCHOS_COMPLEX

template <>
struct TypeMap<Superlu,std::complex<float> >
{
  static SLU::Dtype_t dtype;
  typedef SLU::C::complex convert_type; // to create array before calling superlu
  typedef Kokkos::complex<float> type;
  typedef float magnitude_type;
};


template <>
struct TypeMap<Superlu,std::complex<double> >
{
  static SLU::Dtype_t dtype;
  typedef SLU::Z::doublecomplex convert_type; // to create array before calling superlu
  typedef Kokkos::complex<double> type;
  typedef double magnitude_type;
};


template <>
struct TypeMap<Superlu,Kokkos::complex<float> >
{
  static SLU::Dtype_t dtype;
  typedef SLU::C::complex convert_type; // to create array before calling superlu
  typedef Kokkos::complex<float> type;
  typedef float magnitude_type;
};


template <>
struct TypeMap<Superlu,Kokkos::complex<double> >
{
  static SLU::Dtype_t dtype;
  typedef SLU::Z::doublecomplex convert_type; // to create array before calling superlu
  typedef Kokkos::complex<double> type;
  typedef double magnitude_type;
};


#endif  // HAVE_TEUCHOS_COMPLEX

/* \endcond Superlu_type_specializations */


} // end namespace Amesos2

#endif  // AMESOS2_SUPERLU_TYPEMAP_HPP