chizeta/cnorxz
chizeta/cnorxz
0
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity

WIP: rank array member function implementations

Browse source
This commit is contained in:
Christian Zimmermann 2024-04-14 01:45:15 +02:00
parent 97f28975d5
commit fae345f455
1 changed files with 27 additions and 10 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

33
src/opt/mpi/include/rarray.cc.h
View file

@ -72,30 +72,42 @@ namespace CNORXZ
template <class... Indices> template <class... Indices>
T RCArray<T>::operator[](const SPack<Indices...>& pack) const T RCArray<T>::operator[](const SPack<Indices...>& pack) const
{ {
CXZ_ERROR("not implemented"); typedef typename std::remove_reference<decltype(*pack[CSizeT<0>{}])>::type I0;
return T(); if constexpr(is_rank_index<I0>::value){
return *(begin() + pack.lex());
}
else {
return (*mA)[pack];
}
} }
template <typename T> template <typename T>
template <class... Indices> template <class... Indices>
T RCArray<T>::at(const SPack<Indices...>& pack) const T RCArray<T>::at(const SPack<Indices...>& pack) const
{ {
CXZ_ERROR("not implemented"); typedef typename std::remove_reference<decltype(*pack[CSizeT<0>{}])>::type I0;
return T(); if constexpr(is_rank_index<I0>::value){
CXZ_ASSERT(pack[CSizeT<0>{}]->rank() == getNumRanks(),
"rank number out of scope");
return *(begin() + pack.lex());
}
else {
return mA->at(pack);
}
} }
template <typename T> template <typename T>
T RCArray<T>::operator[](const DPack& pack) const T RCArray<T>::operator[](const DPack& pack) const
{ {
CXZ_ERROR("not implemented"); // TODO: assert that none of the indices is rank index
return T(); return (*mA)[pack];
} }
template <typename T> template <typename T>
T RCArray<T>::at(const DPack& pack) const T RCArray<T>::at(const DPack& pack) const
{ {
CXZ_ERROR("not implemented"); // TODO: assert that none of the indices is rank index
return T(); return mA->at(pack);
} }
template <typename T> template <typename T>
@ -103,9 +115,14 @@ namespace CNORXZ
Sptr<CArrayBase<T>> RCArray<T>::sl(const IndexInterface<I,M>& begin, Sptr<CArrayBase<T>> RCArray<T>::sl(const IndexInterface<I,M>& begin,
const IndexInterface<I,M>& end) const const IndexInterface<I,M>& end) const
{ {
if constexpr(is_rank_index<I>::value){
CXZ_ERROR("not implemented"); CXZ_ERROR("not implemented");
return nullptr; return nullptr;
} }
else {
return mA->sl(begin, end);
}
}
template <typename T> template <typename T>
template <class Index> template <class Index>