/* ************************************************************************ * Copyright (c) 2018 Advanced Micro Devices, Inc. * * Permission is hereby granted, free of charge, to any person obtaining a copy * of this software and associated documentation files (the "Software"), to deal * in the Software without restriction, including without limitation the rights * to use, copy, modify, merge, publish, distribute, sublicense, and/or sell * copies of the Software, and to permit persons to whom the Software is * furnished to do so, subject to the following conditions: * * The above copyright notice and this permission notice shall be included in * all copies or substantial portions of the Software. * * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY, * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE * AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, * OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN * THE SOFTWARE. * * ************************************************************************ */ #ifndef ROCALUTION_HOST_VECTOR_HPP_ #define ROCALUTION_HOST_VECTOR_HPP_ #include "../base_matrix.hpp" #include "../base_stencil.hpp" #include "../base_vector.hpp" #include namespace rocalution { template class LocalVector; template class HostVector : public BaseVector { public: HostVector(); HostVector(const Rocalution_Backend_Descriptor local_backend); virtual ~HostVector(); virtual void Info(void) const; virtual bool Check(void) const; virtual void Allocate(int n); virtual void SetDataPtr(ValueType** ptr, int size); virtual void LeaveDataPtr(ValueType** ptr); virtual void Clear(void); virtual void Zeros(void); virtual void Ones(void); virtual void SetValues(ValueType val); virtual void SetRandomUniform(unsigned long long seed, ValueType a, ValueType b); virtual void SetRandomNormal(unsigned long long seed, ValueType mean, ValueType var); virtual void CopyFrom(const BaseVector& vec); virtual void CopyFromFloat(const BaseVector& vec); virtual void CopyFromDouble(const BaseVector& vec); virtual void CopyTo(BaseVector* vec) const; virtual void CopyFrom(const BaseVector& src, int src_offset, int dst_offset, int size); virtual void CopyFromPermute(const BaseVector& src, const BaseVector& permutation); virtual void CopyFromPermuteBackward(const BaseVector& src, const BaseVector& permutation); virtual void CopyFromData(const ValueType* data); virtual void CopyToData(ValueType* data) const; virtual void Permute(const BaseVector& permutation); virtual void PermuteBackward(const BaseVector& permutation); virtual bool Restriction(const BaseVector& vec_fine, const BaseVector& map); virtual bool Prolongation(const BaseVector& vec_coarse, const BaseVector& map); /// Read vector from ASCII file void ReadFileASCII(const std::string filename); /// Write vector to ASCII file void WriteFileASCII(const std::string filename) const; /// Read vector from binary file void ReadFileBinary(const std::string filename); /// Write vector to binary file void WriteFileBinary(const std::string filename) const; // this = this + alpha*x virtual void AddScale(const BaseVector& x, ValueType alpha); // this = alpha*this + x virtual void ScaleAdd(ValueType alpha, const BaseVector& x); // this = alpha*this + x*beta virtual void ScaleAddScale(ValueType alpha, const BaseVector& x, ValueType beta); virtual void ScaleAddScale(ValueType alpha, const BaseVector& x, ValueType beta, int src_offset, int dst_offset, int size); // this = alpha*this + x*beta + y*gamma virtual void ScaleAdd2(ValueType alpha, const BaseVector& x, ValueType beta, const BaseVector& y, ValueType gamma); // this = alpha*this virtual void Scale(ValueType alpha); // this^T x virtual ValueType Dot(const BaseVector& x) const; // this^T x virtual ValueType DotNonConj(const BaseVector& x) const; // srqt(this^T this) virtual ValueType Norm(void) const; // reduce vector virtual ValueType Reduce(void) const; // Compute sum of absolute values of this virtual ValueType Asum(void) const; // Compute absolute value of this virtual int Amax(ValueType& value) const; // point-wise multiplication virtual void PointWiseMult(const BaseVector& x); virtual void PointWiseMult(const BaseVector& x, const BaseVector& y); virtual void Power(double power); // set index array virtual void SetIndexArray(int size, const int* index); // get index values virtual void GetIndexValues(ValueType* values) const; // set index values virtual void SetIndexValues(const ValueType* values); // get continuous index values virtual void GetContinuousValues(int start, int end, ValueType* values) const; // set continuous index values virtual void SetContinuousValues(int start, int end, const ValueType* values); // get coarse boundary mapping virtual void ExtractCoarseMapping( int start, int end, const int* index, int nc, int* size, int* map) const; // get coarse boundary index virtual void ExtractCoarseBoundary( int start, int end, const int* index, int nc, int* size, int* boundary) const; private: ValueType* vec_; int* index_array_; // for [] operator in LocalVector friend class LocalVector; friend class HostVector; friend class HostVector; friend class HostVector>; friend class HostVector>; friend class HostVector; friend class HostMatrix; friend class HostMatrixCSR; friend class HostMatrixCOO; friend class HostMatrixDIA; friend class HostMatrixELL; friend class HostMatrixHYB; friend class HostMatrixDENSE; friend class HostMatrixMCSR; friend class HostMatrixBCSR; friend class HostMatrixCOO; friend class HostMatrixCOO; friend class HostMatrixCOO>; friend class HostMatrixCOO>; friend class HostMatrixCSR; friend class HostMatrixCSR; friend class HostMatrixCSR>; friend class HostMatrixCSR>; friend class HostMatrixDENSE; friend class HostMatrixDENSE; friend class HIPAcceleratorVector; friend class HostStencil; friend class HostStencilLaplace2D; }; } // namespace rocalution #endif // ROCALUTION_HOST_VECTOR_HPP_