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

Added TODO for perf, renamed to bvector.

Browse Source
This commit is contained in:
Kjetil Olsen Lye 2023-05-30 09:54:22 +02:00
parent 0ca3c4f959
commit b4b1a7b77f
1 changed files with 14 additions and 12 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -150,45 +150,47 @@ public:
/** /**
* @brief copyFromHost copies data from a Dune::BlockVector * @brief copyFromHost copies data from a Dune::BlockVector
* @param vector the vector to copy from * @param bvector the vector to copy from
* *
* @note This does synchronous transfer. * @note This does synchronous transfer.
* @note This assumes that the size of this vector is equal to the dim of the input vector. * @note This assumes that the size of this vector is equal to the dim of the input vector.
*/ */
template <int BlockDimension> template <int BlockDimension>
void copyFromHost(const Dune::BlockVector<Dune::FieldVector<T, BlockDimension>>& vector) void copyFromHost(const Dune::BlockVector<Dune::FieldVector<T, BlockDimension>>& bvector)
{ {
if (detail::to_size_t(m_numberOfElements) != vector.dim()) { // TODO: [perf] vector.dim() can be replaced by bvector.N() * BlockDimension
if (detail::to_size_t(m_numberOfElements) != bvector.dim()) {
OPM_THROW(std::runtime_error, OPM_THROW(std::runtime_error,
fmt::format("Given incompatible vector size. CuVector has size {}, \n" fmt::format("Given incompatible vector size. CuVector has size {}, \n"
"however, BlockVector has N() = {}, and dim = {}.", "however, BlockVector has N() = {}, and dim = {}.",
m_numberOfElements, m_numberOfElements,
vector.N(), bvector.N(),
vector.dim())); bvector.dim()));
} }
const auto dataPointer = static_cast<const T*>(&(vector[0][0])); const auto dataPointer = static_cast<const T*>(&(bvector[0][0]));
copyFromHost(dataPointer, m_numberOfElements); copyFromHost(dataPointer, m_numberOfElements);
} }
/** /**
* @brief copyToHost copies data to a Dune::BlockVector * @brief copyToHost copies data to a Dune::BlockVector
* @param vector the vector to copy to * @param bvector the vector to copy to
* *
* @note This does synchronous transfer. * @note This does synchronous transfer.
* @note This assumes that the size of this vector is equal to the dim of the input vector. * @note This assumes that the size of this vector is equal to the dim of the input vector.
*/ */
template <int BlockDimension> template <int BlockDimension>
void copyToHost(Dune::BlockVector<Dune::FieldVector<T, BlockDimension>>& vector) const void copyToHost(Dune::BlockVector<Dune::FieldVector<T, BlockDimension>>& bvector) const
{ {
if (detail::to_size_t(m_numberOfElements) != vector.dim()) { // TODO: [perf] vector.dim() can be replaced by bvector.N() * BlockDimension
if (detail::to_size_t(m_numberOfElements) != bvector.dim()) {
OPM_THROW(std::runtime_error, OPM_THROW(std::runtime_error,
fmt::format("Given incompatible vector size. CuVector has size {},\n however, the BlockVector " fmt::format("Given incompatible vector size. CuVector has size {},\n however, the BlockVector "
"has has N() = {}, and dim() = {}.", "has has N() = {}, and dim() = {}.",
m_numberOfElements, m_numberOfElements,
vector.N(), bvector.N(),
vector.dim())); bvector.dim()));
} }
const auto dataPointer = static_cast<T*>(&(vector[0][0])); const auto dataPointer = static_cast<T*>(&(bvector[0][0]));
copyToHost(dataPointer, m_numberOfElements); copyToHost(dataPointer, m_numberOfElements);
} }