OilfieldToolsNet/opm-simulators
4
0
Fork 0
mirror of https://github.com/OPM/opm-simulators.git synced 2025-02-25 18:55:30 -06:00
Code Issues Packages Projects Releases Wiki Activity

Updated documentation to reflect limitations on int.

Browse Source
This commit is contained in:
Kjetil Olsen Lye 2023-05-16 10:01:39 +02:00
parent b4aa427bbe
commit 70310160c0
1 changed files with 31 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

37
opm/simulators/linalg/cuistl/CuVector.hpp
View File

@ -34,10 +34,13 @@ namespace Opm::cuistl
/** /**
* @brief The CuVector class is a simple (arithmetic) vector class for the GPU. * @brief The CuVector class is a simple (arithmetic) vector class for the GPU.
* *
* @note we currently only support simple raw primitives for T (double and float) * @note we currently only support simple raw primitives for T (double, float and int)
*
* @note We currently only support arithmetic operations on double and float.
* *
* @note this vector has no notion of block size. The user is responsible for allocating * @note this vector has no notion of block size. The user is responsible for allocating
* the correct number of primitives (double or floats) * the correct number of primitives (double or floats)
*
* Example usage: * Example usage:
* *
* @code{.cpp} * @code{.cpp}
@ -230,6 +233,8 @@ public:
* @param scalar the scalar to with which to multiply every element * @param scalar the scalar to with which to multiply every element
* *
* @note This operation is asynchronous. * @note This operation is asynchronous.
*
* @note int is not supported
*/ */
CuVector<T>& operator*=(const T& scalar); CuVector<T>& operator*=(const T& scalar);
@ -278,13 +283,33 @@ public:
*/ */
T two_norm() const; T two_norm() const;
//! Computes the dot product sum_i this[indexSet[i]] * other[indexSet[i]] /**
* Computes the dot product sum_i this[indexSet[i]] * other[indexSet[i]]
*
* @note int is not supported
*/
T dot(const CuVector<T>& other, const CuVector<int>& indexSet, CuVector<T>& buffer) const; T dot(const CuVector<T>& other, const CuVector<int>& indexSet, CuVector<T>& buffer) const;
//! Computes the norm sqrt(sum_i this[indexSet[i]] * this[indexSet[i]])
/**
* Computes the norm sqrt(sum_i this[indexSet[i]] * this[indexSet[i]])
*
* @note int is not supported
*/
T two_norm(const CuVector<int>& indexSet, CuVector<T>& buffer) const; T two_norm(const CuVector<int>& indexSet, CuVector<T>& buffer) const;
//! Computes the dot product sum_i this[indexSet[i]] * other[indexSet[i]]
/**
* Computes the dot product sum_i this[indexSet[i]] * other[indexSet[i]]
*
* @note int is not supported
*/
T dot(const CuVector<T>& other, const CuVector<int>& indexSet) const; T dot(const CuVector<T>& other, const CuVector<int>& indexSet) const;
//! Computes the norm sqrt(sum_i this[indexSet[i]] * this[indexSet[i]])
/**
* Computes the norm sqrt(sum_i this[indexSet[i]] * this[indexSet[i]])
*
* @note int is not supported
*/
T two_norm(const CuVector<int>& indexSet) const; T two_norm(const CuVector<int>& indexSet) const;
@ -338,7 +363,7 @@ private:
T* m_dataOnDevice = nullptr; T* m_dataOnDevice = nullptr;
// Note that we store this as int to make sure we are always cublas compatible. // Note that we store this as int to make sure we are always cublas compatible.
// This gives the added benifit that a size_t to int conversion error occurs during construction. // This gives the added benefit that a size_t to int conversion error occurs during construction.
const int m_numberOfElements; const int m_numberOfElements;
detail::CuBlasHandle& m_cuBlasHandle; detail::CuBlasHandle& m_cuBlasHandle;