46 #ifndef XPETRA_MULTIVECTOR_DECL_HPP
47 #define XPETRA_MULTIVECTOR_DECL_HPP
49 #include <Teuchos_LabeledObject.hpp>
50 #include <Teuchos_DataAccess.hpp>
51 #include <Teuchos_BLAS_types.hpp>
52 #include <Teuchos_Range1D.hpp>
60 #include <Kokkos_Core.hpp>
61 #include <Kokkos_DualView.hpp>
62 #include <Kokkos_ArithTraits.hpp>
66 #ifndef DOXYGEN_SHOULD_SKIP_THIS
68 template <
class S,
class LO,
class GO,
class N>
72 template <
class Scalar,
75 class Node = Tpetra::KokkosClassic::DefaultNode::DefaultNodeType>
77 :
public DistObject<Scalar, LocalOrdinal, GlobalOrdinal, Node> {
108 virtual void replaceGlobalValue(GlobalOrdinal globalRow,
size_t vectorIndex,
const Scalar& value) = 0;
111 virtual void sumIntoGlobalValue(GlobalOrdinal globalRow,
size_t vectorIndex,
const Scalar& value) = 0;
114 virtual void replaceLocalValue(LocalOrdinal myRow,
size_t vectorIndex,
const Scalar& value) = 0;
117 virtual void sumIntoLocalValue(LocalOrdinal myRow,
size_t vectorIndex,
const Scalar& value) = 0;
120 virtual void putScalar(
const Scalar& value) = 0;
128 virtual Teuchos::RCP<const Vector<Scalar, LocalOrdinal, GlobalOrdinal, Node>>
getVector(
size_t j)
const = 0;
131 virtual Teuchos::RCP<Vector<Scalar, LocalOrdinal, GlobalOrdinal, Node>>
getVectorNonConst(
size_t j) = 0;
134 virtual Teuchos::ArrayRCP<const Scalar>
getData(
size_t j)
const = 0;
154 virtual void scale(
const Scalar& alpha) = 0;
157 virtual void scale(Teuchos::ArrayView<const Scalar> alpha) = 0;
165 const Scalar& gamma) = 0;
168 virtual void norm1(
const Teuchos::ArrayView<
typename Teuchos::ScalarTraits<Scalar>::magnitudeType>& norms)
const = 0;
171 virtual void norm2(
const Teuchos::ArrayView<
typename Teuchos::ScalarTraits<Scalar>::magnitudeType>& norms)
const = 0;
174 virtual void normInf(
const Teuchos::ArrayView<
typename Teuchos::ScalarTraits<Scalar>::magnitudeType>& norms)
const = 0;
178 virtual void meanValue(
const Teuchos::ArrayView<Scalar>& means)
const = 0;
181 virtual void multiply(Teuchos::ETransp transA,
182 Teuchos::ETransp transB,
186 const Scalar& beta) = 0;
218 Scalar scalarThis) = 0;
246 virtual void describe(Teuchos::FancyOStream& out,
const Teuchos::EVerbosityLevel verbLevel = Teuchos::Describable::verbLevel_default)
const = 0;
256 virtual void setSeed(
unsigned int seed) = 0;
259 virtual void randomize(
bool bUseXpetraImplementation =
false) = 0;
262 virtual void randomize(
const Scalar& minVal,
const Scalar& maxVal,
bool bUseXpetraImplementation =
false) = 0;
271 using dual_view_type = Kokkos::DualView<impl_scalar_type**, Kokkos::LayoutStride, typename node_type::device_type, Kokkos::MemoryUnmanaged>;
272 using dual_view_type_const = Kokkos::DualView<const impl_scalar_type**, Kokkos::LayoutStride, typename node_type::device_type, Kokkos::MemoryUnmanaged>;
277 throw std::runtime_error(
"Dummy function getHostLocalView(Access::ReadOnlyStruct), should be overwritten at" + std::string(__FILE__) +
":" + std::to_string(__LINE__));
279 typename dual_view_type::t_host_um test;
281 TEUCHOS_UNREACHABLE_RETURN(test);
285 throw std::runtime_error(
"Dummy function getDeviceLocalView(Access::ReadOnlyStruct), should be overwritten at" + std::string(__FILE__) +
":" + std::to_string(__LINE__));
287 typename dual_view_type::t_dev_um test;
289 TEUCHOS_UNREACHABLE_RETURN(test);
293 throw std::runtime_error(
"Dummy function getHostLocalView(Access::OverwriteAllStruct), should be overwritten at" + std::string(__FILE__) +
":" + std::to_string(__LINE__));
295 typename dual_view_type::t_host_um test;
297 TEUCHOS_UNREACHABLE_RETURN(test);
301 throw std::runtime_error(
"Dummy function getDeviceLocalView(Access::OverwriteAllStruct), should be overwritten at" + std::string(__FILE__) +
":" + std::to_string(__LINE__));
303 typename dual_view_type::t_dev_um test;
305 TEUCHOS_UNREACHABLE_RETURN(test);
309 throw std::runtime_error(
"Dummy function getHostLocalView(Access::ReadWriteStruct), should be overwritten at" + std::string(__FILE__) +
":" + std::to_string(__LINE__));
311 typename dual_view_type::t_host_um test;
313 TEUCHOS_UNREACHABLE_RETURN(test);
317 throw std::runtime_error(
"Dummy function getDeviceLocalView(Access::ReadWriteStruct), should be overwritten at" + std::string(__FILE__) +
":" + std::to_string(__LINE__));
319 typename dual_view_type::t_dev_um test;
321 TEUCHOS_UNREACHABLE_RETURN(test);
338 #define XPETRA_MULTIVECTOR_SHORT
340 #endif // XPETRA_MULTIVECTOR_DECL_HPP
virtual void replaceMap(const RCP< const Map< LocalOrdinal, GlobalOrdinal, Node >> &map)=0
virtual void norm2(const Teuchos::ArrayView< typename Teuchos::ScalarTraits< Scalar >::magnitudeType > &norms) const =0
virtual dual_view_type::t_dev_const_um getDeviceLocalView(Access::ReadOnlyStruct) const
virtual void normInf(const Teuchos::ArrayView< typename Teuchos::ScalarTraits< Scalar >::magnitudeType > &norms) const =0
Compute Inf-norm of each vector in multi-vector.
virtual dual_view_type::t_host_um getHostLocalView(Access::ReadWriteStruct) const
virtual std::string description() const =0
A simple one-line description of this object.
virtual void randomize(bool bUseXpetraImplementation=false)=0
Set multi-vector values to random numbers.
virtual Teuchos::RCP< Vector< Scalar, LocalOrdinal, GlobalOrdinal, Node > > getVectorNonConst(size_t j)=0
Return a Vector which is a nonconst view of column j.
virtual void sumIntoGlobalValue(GlobalOrdinal globalRow, size_t vectorIndex, const Scalar &value)=0
Add value to existing value, using global (row) index.
Kokkos::DualView< const impl_scalar_type **, Kokkos::LayoutStride, typename node_type::device_type, Kokkos::MemoryUnmanaged > dual_view_type_const
virtual void putScalar(const Scalar &value)=0
Set all values in the multivector with the given value.
virtual void setSeed(unsigned int seed)=0
Set seed for Random function.
virtual void Xpetra_randomize()
Set multi-vector values to random numbers. XPetra implementation.
MultiVector< Scalar, LocalOrdinal, GlobalOrdinal, Node > & operator=(const MultiVector< Scalar, LocalOrdinal, GlobalOrdinal, Node > &rhs)
Assignment operator: Does a deep copy.
virtual void assign(const MultiVector< Scalar, LocalOrdinal, GlobalOrdinal, Node > &rhs)=0
Implementation of the assignment operator (operator=); does a deep copy.
virtual void reciprocal(const MultiVector< Scalar, LocalOrdinal, GlobalOrdinal, Node > &A)=0
Put element-wise reciprocal values of input Multi-vector in target, this(i,j) = 1/A(i,j).
virtual void update(const Scalar &alpha, const MultiVector< Scalar, LocalOrdinal, GlobalOrdinal, Node > &A, const Scalar &beta)=0
Update multi-vector values with scaled values of A, this = beta*this + alpha*A.
virtual void sumIntoLocalValue(LocalOrdinal myRow, size_t vectorIndex, const Scalar &value)=0
Add value to existing value, using local (row) index.
virtual dual_view_type::t_dev_um getDeviceLocalView(Access::ReadWriteStruct) const
virtual bool isSameSize(const MultiVector< Scalar, LocalOrdinal, GlobalOrdinal, Node > &vec) const =0
virtual void describe(Teuchos::FancyOStream &out, const Teuchos::EVerbosityLevel verbLevel=Teuchos::Describable::verbLevel_default) const =0
Print the object with the given verbosity level to a FancyOStream.
virtual ~MultiVector()
Destructor.
virtual Teuchos::ArrayRCP< const Scalar > getData(size_t j) const =0
Const view of the local values in a particular vector of this multivector.
virtual void scale(const Scalar &alpha)=0
Scale the current values of a multi-vector, this = alpha*this.
virtual void elementWiseMultiply(Scalar scalarAB, const Vector< Scalar, LocalOrdinal, GlobalOrdinal, Node > &A, const MultiVector< Scalar, LocalOrdinal, GlobalOrdinal, Node > &B, Scalar scalarThis)=0
Multiply a Vector A elementwise by a MultiVector B.
typename dual_view_type::t_dev::execution_space dev_execution_space
virtual void norm1(const Teuchos::ArrayView< typename Teuchos::ScalarTraits< Scalar >::magnitudeType > &norms) const =0
Compute 1-norm of each vector in multi-vector.
virtual void multiply(Teuchos::ETransp transA, Teuchos::ETransp transB, const Scalar &alpha, const MultiVector< Scalar, LocalOrdinal, GlobalOrdinal, Node > &A, const MultiVector< Scalar, LocalOrdinal, GlobalOrdinal, Node > &B, const Scalar &beta)=0
Matrix-matrix multiplication: this = beta*this + alpha*op(A)*op(B).
LocalOrdinal local_ordinal_type
virtual size_t getLocalLength() const =0
Local number of rows on the calling process.
virtual Teuchos::ArrayRCP< Scalar > getDataNonConst(size_t j)=0
View of the local values in a particular vector of this multivector.
size_t global_size_t
Global size_t object.
virtual global_size_t getGlobalLength() const =0
Global number of rows in the multivector.
Kokkos::DualView< impl_scalar_type **, Kokkos::LayoutStride, typename node_type::device_type, Kokkos::MemoryUnmanaged > dual_view_type
virtual void meanValue(const Teuchos::ArrayView< Scalar > &means) const =0
GlobalOrdinal global_ordinal_type
virtual dual_view_type::t_host_um getHostLocalView(Access::OverwriteAllStruct) const
typename dual_view_type::host_mirror_space host_execution_space
virtual void abs(const MultiVector< Scalar, LocalOrdinal, GlobalOrdinal, Node > &A)=0
Put element-wise absolute values of input Multi-vector in target: A = abs(this).
virtual dual_view_type::t_dev_um getDeviceLocalView(Access::OverwriteAllStruct) const
virtual dual_view_type::t_host_const_um getHostLocalView(Access::ReadOnlyStruct) const
virtual void dot(const MultiVector< Scalar, LocalOrdinal, GlobalOrdinal, Node > &A, const Teuchos::ArrayView< Scalar > &dots) const =0
Compute dot product of each corresponding pair of vectors, dots[i] = this[i].dot(A[i]).
typename Kokkos::ArithTraits< int >::val_type impl_scalar_type
virtual void replaceGlobalValue(GlobalOrdinal globalRow, size_t vectorIndex, const Scalar &value)=0
Replace value, using global (row) index.
virtual void replaceLocalValue(LocalOrdinal myRow, size_t vectorIndex, const Scalar &value)=0
Replace value, using local (row) index.
virtual size_t getNumVectors() const =0
Number of columns in the multivector.
virtual Teuchos::RCP< const Vector< Scalar, LocalOrdinal, GlobalOrdinal, Node > > getVector(size_t j) const =0
Return a Vector which is a const view of column j.