Amesos2 - Direct Sparse Solver Interfaces Version of the Day
Amesos2_TpetraCrsMatrix_MatrixAdapter_def.hpp
1// @HEADER
2//
3// ***********************************************************************
4//
5// Amesos2: Templated Direct Sparse Solver Package
6// Copyright 2011 Sandia Corporation
7//
8// Under the terms of Contract DE-AC04-94AL85000 with Sandia Corporation,
9// the U.S. Government retains certain rights in this software.
10//
11// Redistribution and use in source and binary forms, with or without
12// modification, are permitted provided that the following conditions are
13// met:
14//
15// 1. Redistributions of source code must retain the above copyright
16// notice, this list of conditions and the following disclaimer.
17//
18// 2. Redistributions in binary form must reproduce the above copyright
19// notice, this list of conditions and the following disclaimer in the
20// documentation and/or other materials provided with the distribution.
21//
22// 3. Neither the name of the Corporation nor the names of the
23// contributors may be used to endorse or promote products derived from
24// this software without specific prior written permission.
25//
26// THIS SOFTWARE IS PROVIDED BY SANDIA CORPORATION "AS IS" AND ANY
27// EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
28// IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
29// PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL SANDIA CORPORATION OR THE
30// CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL,
31// EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO,
32// PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR
33// PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF
34// LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING
35// NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS
36// SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
37//
38// Questions? Contact Michael A. Heroux (maherou@sandia.gov)
39//
40// ***********************************************************************
41//
42// @HEADER
43
44
45#ifndef AMESOS2_TPETRACRSMATRIX_MATRIXADAPTER_DEF_HPP
46#define AMESOS2_TPETRACRSMATRIX_MATRIXADAPTER_DEF_HPP
47
49#include "Amesos2_TpetraRowMatrix_AbstractMatrixAdapter_def.hpp"
50#include "Amesos2_MatrixAdapter_def.hpp"
51
52namespace Amesos2 {
53
54 template <typename Scalar,
55 typename LocalOrdinal,
56 typename GlobalOrdinal,
57 typename Node>
58 ConcreteMatrixAdapter<
59 Tpetra::CrsMatrix<Scalar,
60 LocalOrdinal,
61 GlobalOrdinal,
62 Node>
63 >::ConcreteMatrixAdapter(Teuchos::RCP<matrix_t> m)
64 : AbstractConcreteMatrixAdapter<Tpetra::RowMatrix<Scalar,
65 LocalOrdinal,
66 GlobalOrdinal,
67 Node>,
68 Tpetra::CrsMatrix<Scalar,
69 LocalOrdinal,
70 GlobalOrdinal,
71 Node> >(m) // with implicit cast
72 {}
73
74 template <typename Scalar,
75 typename LocalOrdinal,
76 typename GlobalOrdinal,
77 typename Node>
78 Teuchos::RCP<const MatrixAdapter<Tpetra::CrsMatrix<Scalar,LocalOrdinal,GlobalOrdinal,Node> > >
79 ConcreteMatrixAdapter<
80 Tpetra::CrsMatrix<Scalar, LocalOrdinal, GlobalOrdinal, Node>
81 >::get_impl(const Teuchos::Ptr<const Tpetra::Map<local_ordinal_t,global_ordinal_t,node_t> > map, EDistribution distribution) const
82 {
83 using Teuchos::RCP;
84 using Teuchos::rcp;
85 using Teuchos::rcpFromPtr;
86 typedef Tpetra::Import<local_ordinal_t, global_ordinal_t, node_t> import_t;
87
88 RCP<import_t> importer =
89 rcp (new import_t (this->getRowMap(), rcpFromPtr (map)));
90
91 RCP<matrix_t> t_mat;
92
93 t_mat = Tpetra::importAndFillCompleteCrsMatrix<matrix_t>( (this->mat_), *importer ); // DomainMap, RangeMap, params inputs default to Teuchos::null
94
95 // Case for non-contiguous GIDs
96 if ( distribution == CONTIGUOUS_AND_ROOTED ) {
97
98 auto myRank = map->getComm()->getRank();
99
100 auto local_matrix = t_mat->getLocalMatrixDevice();
101 const size_t global_num_contiguous_entries = t_mat->getGlobalNumRows();
102 const size_t local_num_contiguous_entries = (myRank == 0) ? t_mat->getGlobalNumRows() : 0;
103
104 //create maps
105 typedef Tpetra::Map< local_ordinal_t, global_ordinal_t, node_t> contiguous_map_type;
106 RCP<const contiguous_map_type> contiguousRowMap = rcp( new contiguous_map_type(global_num_contiguous_entries, local_num_contiguous_entries, 0, (t_mat->getComm() ) ) );
107 RCP<const contiguous_map_type> contiguousColMap = rcp( new contiguous_map_type(global_num_contiguous_entries, local_num_contiguous_entries, 0, (t_mat->getComm() ) ) );
108 RCP<const contiguous_map_type> contiguousDomainMap = rcp( new contiguous_map_type(global_num_contiguous_entries, local_num_contiguous_entries, 0, (t_mat->getComm() ) ) );
109 RCP<const contiguous_map_type> contiguousRangeMap = rcp( new contiguous_map_type(global_num_contiguous_entries, local_num_contiguous_entries, 0, (t_mat->getComm() ) ) );
110
111 RCP<matrix_t> contiguous_t_mat = rcp( new matrix_t(contiguousRowMap, contiguousColMap, local_matrix) );
112 contiguous_t_mat->resumeFill();
113 contiguous_t_mat->expertStaticFillComplete(contiguousDomainMap, contiguousRangeMap);
114
115 return rcp (new ConcreteMatrixAdapter<matrix_t> (contiguous_t_mat));
116 } //end if not contiguous
117
118 return rcp (new ConcreteMatrixAdapter<matrix_t> (t_mat));
119 }
120
121} // end namespace Amesos2
122
123#endif // AMESOS2_TPETRACRSMATRIX_MATRIXADAPTER_DEF_HPP
Specialization of the ConcreteMatrixAdapter for Tpetra::CrsMatrix. Inherits all its functionality fro...