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