10 #ifndef TPETRA_VECTOR_DEF_HPP
11 #define TPETRA_VECTOR_DEF_HPP
16 #include "Tpetra_MultiVector.hpp"
18 #include "KokkosCompat_View.hpp"
19 #include "KokkosBlas1_nrm2w_squared.hpp"
20 #include "Teuchos_CommHelpers.hpp"
24 template <
class Scalar,
class LocalOrdinal,
class GlobalOrdinal,
class Node>
29 template <
class Scalar,
class LocalOrdinal,
class GlobalOrdinal,
class Node>
31 Vector(
const Teuchos::RCP<const map_type>& map,
35 template <
class Scalar,
class LocalOrdinal,
class GlobalOrdinal,
class Node>
38 const Teuchos::DataAccess copyOrView)
41 template <
class Scalar,
class LocalOrdinal,
class GlobalOrdinal,
class Node>
44 const Teuchos::RCP<const map_type>& map,
48 template <
class Scalar,
class LocalOrdinal,
class GlobalOrdinal,
class Node>
50 Vector(
const Teuchos::RCP<const map_type>& map,
51 const Teuchos::ArrayView<const Scalar>& values)
52 :
base_type(map, values, values.size(), 1) {}
54 template <
class Scalar,
class LocalOrdinal,
class GlobalOrdinal,
class Node>
56 Vector(
const Teuchos::RCP<const map_type>& map,
60 template <
class Scalar,
class LocalOrdinal,
class GlobalOrdinal,
class Node>
62 Vector(
const Teuchos::RCP<const map_type>& map,
67 template <
class Scalar,
class LocalOrdinal,
class GlobalOrdinal,
class Node>
69 Vector(
const Teuchos::RCP<const map_type>& map,
73 template <
class Scalar,
class LocalOrdinal,
class GlobalOrdinal,
class Node>
79 template <
class Scalar,
class LocalOrdinal,
class GlobalOrdinal,
class Node>
82 this->base_type::replaceGlobalValue(globalRow, 0, value);
85 template <
class Scalar,
class LocalOrdinal,
class GlobalOrdinal,
class Node>
90 this->base_type::sumIntoGlobalValue(globalRow, 0, value, atomic);
93 template <
class Scalar,
class LocalOrdinal,
class GlobalOrdinal,
class Node>
96 this->base_type::replaceLocalValue(myRow, 0, value);
99 template <
class Scalar,
class LocalOrdinal,
class GlobalOrdinal,
class Node>
104 this->base_type::sumIntoLocalValue(globalRow, 0, value, atomic);
107 template <
class Scalar,
class LocalOrdinal,
class GlobalOrdinal,
class Node>
110 const size_t lda = this->getLocalLength();
111 this->get1dCopy(A, lda);
114 template <
class Scalar,
class LocalOrdinal,
class GlobalOrdinal,
class Node>
119 this->dot(y, Teuchos::arrayView(&result, 1));
123 template <
class Scalar,
class LocalOrdinal,
class GlobalOrdinal,
class Node>
128 this->meanValue(Teuchos::arrayView(&mean, 1));
132 template <
class Scalar,
class LocalOrdinal,
class GlobalOrdinal,
class Node>
137 this->norm1(Teuchos::arrayView(&norm, 1));
141 template <
class Scalar,
class LocalOrdinal,
class GlobalOrdinal,
class Node>
146 this->norm2(Teuchos::arrayView(&norm, 1));
150 template <
class Scalar,
class LocalOrdinal,
class GlobalOrdinal,
class Node>
155 this->normInf(Teuchos::arrayView(&norm, 1));
159 template <
class Scalar,
class LocalOrdinal,
class GlobalOrdinal,
class Node>
162 return this->descriptionImpl(
"Tpetra::Vector");
165 template <
class Scalar,
class LocalOrdinal,
class GlobalOrdinal,
class Node>
168 const Teuchos::EVerbosityLevel verbLevel)
const {
169 this->describeImpl(out,
"Tpetra::Vector", verbLevel);
172 template <
class Scalar,
class LocalOrdinal,
class GlobalOrdinal,
class Node>
173 Teuchos::RCP<const Vector<Scalar, LocalOrdinal, GlobalOrdinal, Node> >
175 offsetView(
const Teuchos::RCP<const map_type>& subMap,
176 const size_t offset)
const {
178 using Kokkos::subview;
182 const size_t newNumRows = subMap->getLocalNumElements();
183 const bool tooManyElts = newNumRows + offset > this->getOrigNumLocalRows();
185 const int myRank = this->getMap()->getComm()->getRank();
186 TEUCHOS_TEST_FOR_EXCEPTION(
187 newNumRows + offset > this->getLocalLength(), std::runtime_error,
188 "Tpetra::Vector::offsetView(NonConst): Invalid input Map. The input "
190 << newNumRows <<
" entries on process " << myRank <<
". "
192 << offset <<
". Yet, the Vector contains only "
193 << this->getOrigNumLocalRows() <<
" rows on this process.");
197 return rcp(
new V(subMap, wrapped_dual_view_type(this->view_, Kokkos::pair<int, int>(offset, offset + newNumRows), ALL())));
200 template <
class Scalar,
class LocalOrdinal,
class GlobalOrdinal,
class Node>
201 Teuchos::RCP<Vector<Scalar, LocalOrdinal, GlobalOrdinal, Node> >
202 Vector<Scalar, LocalOrdinal, GlobalOrdinal, Node>::
203 offsetViewNonConst(
const Teuchos::RCP<const map_type>& subMap,
204 const size_t offset) {
205 typedef Vector<Scalar, LocalOrdinal, GlobalOrdinal, Node> V;
206 return Teuchos::rcp_const_cast<V>(this->offsetView(subMap, offset));
209 template <
class Scalar,
class LocalOrdinal,
class GlobalOrdinal,
class Node>
210 Vector<Scalar, LocalOrdinal, GlobalOrdinal, Node>
215 vec_type dst(src, Teuchos::Copy);
233 #define TPETRA_VECTOR_INSTANT(SCALAR, LO, GO, NODE) \
234 template class Vector<SCALAR, LO, GO, NODE>; \
235 template Vector<SCALAR, LO, GO, NODE> createCopy(const Vector<SCALAR, LO, GO, NODE>& src);
237 #endif // TPETRA_VECTOR_DEF_HPP
Vector()
Default constructor: makes a Vector with no rows or columns.
mag_type norm2() const
Return the two-norm of this Vector.
virtual void describe(Teuchos::FancyOStream &out, const Teuchos::EVerbosityLevel verbLevel=Teuchos::Describable::verbLevel_default) const
Describe this object in a human-readable way to the given output stream.
base_type::mag_type mag_type
Type of a norm result.
Declaration of a function that prints strings from each process.
void get1dCopy(const Teuchos::ArrayView< Scalar > &A) const
Return multi-vector values in user-provided two-dimensional array (using Teuchos memory management cl...
void sumIntoLocalValue(const LocalOrdinal myRow, const Scalar &value, const bool atomic=base_type::useAtomicUpdatesByDefault)
Add value to existing value, using local (row) index.
virtual std::string description() const
Return a one-line description of this object.
MultiVector< ST, LO, GO, NT > createCopy(const MultiVector< ST, LO, GO, NT > &src)
Return a deep copy of the given MultiVector.
dot_type dot(const Vector< Scalar, LocalOrdinal, GlobalOrdinal, Node > &y) const
Return the dot product of this Vector and the input Vector x.
void sumIntoGlobalValue(const GlobalOrdinal globalRow, const Scalar &value, const bool atomic=base_type::useAtomicUpdatesByDefault)
Add value to existing value, using global (row) index.
void replaceLocalValue(const LocalOrdinal myRow, const Scalar &value)
Replace current value at the specified location with specified values.
void replaceGlobalValue(const GlobalOrdinal globalRow, const Scalar &value)
Replace current value at the specified location with specified value.
LocalOrdinal local_ordinal_type
This class' second template parameter; the type of local indices.
A distributed dense vector.
base_type::dual_view_type dual_view_type
Kokkos::DualView specialization used by this class.
Scalar meanValue() const
Compute mean (average) value of this Vector.
mag_type normInf() const
Return the infinity-norm of this Vector.
base_type::dot_type dot_type
Type of an inner ("dot") product result.
mag_type norm1() const
Return the one-norm of this Vector.