Stokhos Package Browser (Single Doxygen Collection) Version of the Day
Loading...
Searching...
No Matches
Amesos2_Basker_MP_Vector.hpp
Go to the documentation of this file.
1// @HEADER
2// ***********************************************************************
3//
4// Stokhos Package
5// Copyright (2009) Sandia Corporation
6//
7// Under terms of Contract DE-AC04-94AL85000, there is a non-exclusive
8// license for use of this work by or on behalf of the U.S. Government.
9//
10// Redistribution and use in source and binary forms, with or without
11// modification, are permitted provided that the following conditions are
12// met:
13//
14// 1. Redistributions of source code must retain the above copyright
15// notice, this list of conditions and the following disclaimer.
16//
17// 2. Redistributions in binary form must reproduce the above copyright
18// notice, this list of conditions and the following disclaimer in the
19// documentation and/or other materials provided with the distribution.
20//
21// 3. Neither the name of the Corporation nor the names of the
22// contributors may be used to endorse or promote products derived from
23// this software without specific prior written permission.
24//
25// THIS SOFTWARE IS PROVIDED BY SANDIA CORPORATION "AS IS" AND ANY
26// EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
27// IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
28// PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL SANDIA CORPORATION OR THE
29// CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL,
30// EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO,
31// PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR
32// PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF
33// LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING
34// NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS
35// SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
36//
37// Questions? Contact Eric T. Phipps (etphipp@sandia.gov).
38//
39// ***********************************************************************
40// @HEADER
41
42#ifndef AMESOS2_BASKER_MP_VECTOR_HPP
43#define AMESOS2_BASKER_MP_VECTOR_HPP
44
45#include "Amesos2_config.h"
46#ifdef HAVE_AMESOS2_BASKER
47
48#include "Amesos2_Basker.hpp"
51
52// Specialization of BASKER_ScalarTraits for MP::Vector
53template <class T> struct BASKER_ScalarTraits;
54template <class S>
55struct BASKER_ScalarTraits< Sacado::MP::Vector<S> > {
56 typedef Sacado::MP::Vector<S> val_type;
57 typedef Kokkos::Details::ArithTraits<val_type> KAT;
58 typedef typename KAT::mag_type magnitudeType;
59 static inline val_type reciprocal(val_type c){ return 1.0/c; }
60 static inline val_type divide(val_type a, val_type b){ return a/b; }
61 static inline magnitudeType approxABS(val_type a) { return KAT::abs(a); }
62 static inline magnitudeType abs(val_type a) { return KAT::abs(a); }
63 static inline bool gt (val_type a, val_type b){ return (a>b); }
64};
65
66namespace Amesos2 {
67
68 // Enable MP::Vector as a valid Scalar type for Basker
69 template <class ST>
70 struct TypeMap< Basker,Sacado::MP::Vector<ST> > {
71 typedef Sacado::MP::Vector<ST> dtype;
72 typedef Sacado::MP::Vector<ST> type;
73 typedef typename Kokkos::Details::ArithTraits< Sacado::MP::Vector<ST> >::mag_type magnitude_type;
74 };
75
76 // Specialize our specialization for create_solver_with_supported_type
77 // to pass the scalar type directly to Basker
78 template < class ST, class LO, class GO, class NO >
79 struct create_mp_vector_solver_impl < Basker, ST, LO, GO, NO > {
81 typedef Tpetra::CrsMatrix<SC,LO,GO,NO> Matrix;
82 typedef Tpetra::MultiVector<SC,LO,GO,NO> Vector;
83 static Teuchos::RCP<Solver<Matrix,Vector> >
84 apply(Teuchos::RCP<const Matrix> A,
85 Teuchos::RCP<Vector> X,
86 Teuchos::RCP<const Vector> B )
87 {
88 ctassert<
89 Meta::is_same<
90 typename MatrixTraits<Matrix>::scalar_t,
91 typename MultiVecAdapter<Vector>::scalar_t
92 >::value
93 > same_scalar_assertion;
94 (void)same_scalar_assertion; // This stops the compiler from warning about unused declared variables
95
96 // If our assertion did not fail, then create and return a new solver
97 typedef Tpetra::CrsMatrix<Sacado::MP::Vector<ST>,LO,GO,NO> Matrix;
98 typedef Tpetra::MultiVector<Sacado::MP::Vector<ST>,LO,GO,NO> Vector;
99 return Teuchos::rcp( new Basker<Matrix,Vector>(A, X, B) );
100 }
101 };
102}
103#endif // HAVE_AMESOS2_BASKER
104
105#endif // AMESOS2_BASKER_MP_VECTOR_HPP
abs(expr.val())
static Teuchos::RCP< Solver< Matrix, Vector > > apply(Teuchos::RCP< const Matrix > A, Teuchos::RCP< Vector > X, Teuchos::RCP< const Vector > B)
Tpetra::CrsMatrix< SC, LO, GO, NO > Matrix
Tpetra::MultiVector< SC, LO, GO, NO > Vector