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

fix in obj_handle + mpi: RArray test works

Browse source
This commit is contained in:
Christian Zimmermann 2024-03-21 01:18:21 +01:00
parent c43698dc10
commit c1acda90f0
10 changed files with 105 additions and 50 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
src/include/base/obj_handle.cc.h
View file

@ -41,7 +41,7 @@ namespace CNORXZ
template <typename T>
ObjHandle<T>& ObjHandle<T>::operator=(ObjHandle&& a)
{
mC = a.mC;
mC = std::move(a.mC);
return *this;
}

2
src/opt/mpi/include/cnorxz_mpi.cc.h
View file

@ -12,3 +12,5 @@
#include "mpi_wrappers.cc.h"
//#include "rop_types.cc.h"
#include "rrange.cc.h"
#include "raindex.cc.h"
#include "rarray.cc.h"

6
src/opt/mpi/include/cnorxz_mpi.h
View file

@ -11,10 +11,10 @@
#include "mpi_base.h"
#include "mpi_wrappers.h"
//#include "raindex.h"
//#include "rarray.h"
//#include "rop_types.h"
#include "rrange.h"
#include "raindex.h"
#include "rarray.h"
//#include "rop_types.h"
#include "typemap.h"
#include "cnorxz_mpi.cc.h"

6
src/opt/mpi/include/mpi_base.h
View file

@ -27,6 +27,12 @@ namespace CNORXZ
template <class IndexI, class IndexK>
class RIndex;
template <typename T>
class RAIndex;
template <typename T>
class RArray;
// wrapper functions

35
src/opt/mpi/include/raindex.cc.h
View file

@ -12,7 +12,10 @@
#ifndef __cxz_mpi_raindex_cc_h__
#define __cxz_mpi_raindex_cc_h__
namespace CNOXRZ
#include <cstring>
#include "raindex.h"
namespace CNORXZ
{
namespace mpi
{
@ -20,7 +23,7 @@ namespace CNOXRZ
RAIndex<T>::RAIndex(const T* loc, const RangePtr& range, SizeT lexpos) :
RIndex<YIndex,YIndex>(range, lexpos),
mLoc(loc),
mCur(i.rank()),
mCur(rank()),
mThisRank(getRankNumber())
{
setBufferSize();
@ -30,9 +33,9 @@ namespace CNOXRZ
template <typename T>
RAIndex<T>::RAIndex(const T* loc, const RIndex<YIndex,YIndex>& i) :
RIndex<YIndex,YIndex>(i)
RIndex<YIndex,YIndex>(i),
mLoc(loc),
mCur(i.rank()),
mCur(rank()),
mThisRank(getRankNumber())
{
setBufferSize();
@ -65,25 +68,25 @@ namespace CNOXRZ
template <typename T>
const T& RAIndex<T>::operator*() const
{
if(mCur != rank()){
setBuffer();
}
if(rank() != mThisRank){
if(mCur != rank()){
setBuffer();
}
return mBuf[local()->pos() % mBufSize];
}
else {
mLoc[local()->pos()];
return mLoc[local()->pos()];
}
}
template <typename T>
const T* RAIndex<T>::operator->() const
{
if(mCur != rank()){
setBuffer();
}
if(rank() != mThisRank){
if(mCur != rank()){
setBuffer();
}
return mBuf + local()->pos() % mBufSize;
return mBuf.data() + local()->pos() % mBufSize;
}
else {
return mLoc + local()->pos();
@ -104,16 +107,20 @@ namespace CNOXRZ
}
template <typename T>
void RAIndex<T>::setBuffer()
void RAIndex<T>::setBuffer() const
{
if(mBuf.size() != mBufSize){
mBuf.resize(mBufSize);
}
// A Bcast alternative with const pointer to source would be better...
std::memcpy(mBuf.data(), mLoc + local()->pos() / mBufSize, mBufSize*sizeof(T));
const T* d = mLoc + (local()->pos() / mBufSize) * mBufSize;
std::memcpy(mBuf.data(), d, mBufSize*sizeof(T));
MPI_Bcast(mBuf.data(), mBufSize*sizeof(T), MPI_BYTE, static_cast<int>(rank()),
MPI_COMM_WORLD );
mCur = rank();
}
} // namespace mpi
} // namespace CNOXRZ
#endif

7
src/opt/mpi/include/raindex.h
View file

@ -13,6 +13,7 @@
#define __cxz_mpi_raindex_h__
#include "cnorxz.h"
#include "rrange.h"
namespace CNORXZ
{
@ -49,11 +50,11 @@ namespace CNORXZ
private:
void setBuffer();
void setBuffer() const;
const T* mLoc = nullptr;
Vector<T> mBuf; // used if iterating over content on different rank
SizeT mCur; // current rank in the buffer
mutable Vector<T> mBuf; // used if iterating over content on different rank
mutable SizeT mCur; // current rank in the buffer
SizeT mBufSize;
SizeT mThisRank;
};

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

@ -12,16 +12,34 @@
#ifndef __cxz_mpi_rarray_cc_h__
#define __cxz_mpi_rarray_cc_h__
#include "rarray.h"
#include "raindex.h"
namespace CNORXZ
{
namespace mpi
{
template <typename T>
RCArray<T>::RCArray(const Sptr<CArrayBase<T>> a, const RangePtr& geom) :
mA(a),
RCArray<T>::RCArray(const RCArray& a) :
mA(a.mA->copy()),
mGeom(a.mGeom),
mGlobal(a.mGlobal)
{}
template <typename T>
RCArray<T>& RCArray<T>::operator=(const RCArray& a)
{
mA = ObjHandle<CArrayBase<T>>(a.mA->copy());
mGeom = a.mGeom;
mGlobal = a.mGlobal;
}
template <typename T>
RCArray<T>::RCArray(const CArrayBase<T>& a, const RangePtr& geom) :
mA(a.copy()),
mGeom(geom),
mGlobal(RRangeFactory(a->range(),mGeom).create())
mGlobal(RRangeFactory(rangeCast<YRange>(a.range()),rangeCast<YRange>(mGeom)).create())
{}
template <typename T>
@ -102,14 +120,16 @@ namespace CNORXZ
inline decltype(auto) RCArray<T>::operator()(const SPack<Indices...>& pack) const
{
CXZ_ERROR("not implemented");
return COpRoot<T,Index>();
//return COpRoot<T,Index>();
return 0;
}
template <typename T>
inline decltype(auto) RCArray<T>::operator()(const DPack& pack) const
{
CXZ_ERROR("not implemented");
return COpRoot<T,Index>();
//return COpRoot<T,Index>();
return 0;
}
template <typename T>
@ -131,27 +151,27 @@ namespace CNORXZ
}
template <typename T>
const_iterator RCArray<T>::begin() const
typename RCArray<T>::const_iterator RCArray<T>::begin() const
{
return const_iterator(mA.data(), mGlobal);
return const_iterator(mA->data(), mGlobal);
}
template <typename T>
const_iterator RCArray<T>::end() const
typename RCArray<T>::const_iterator RCArray<T>::end() const
{
return const_iterator(mA.data(), mGlobal, mGlobal->size());
return const_iterator(mA->data(), mGlobal, mGlobal->size());
}
template <typename T>
const_iterator RCArray<T>::cbegin() const
typename RCArray<T>::const_iterator RCArray<T>::cbegin() const
{
return const_iterator(mA.data(), mGlobal);
return const_iterator(mA->data(), mGlobal);
}
template <typename T>
const_iterator RCArray<T>::cend() const
typename RCArray<T>::const_iterator RCArray<T>::cend() const
{
return const_iterator(mA.data(), mGlobal, mGlobal->size());
return const_iterator(mA->data(), mGlobal, mGlobal->size());
}
template <typename T>

10
src/opt/mpi/include/rarray.h
View file

@ -13,6 +13,7 @@
#define __cxz_mpi_rarray_h__
#include "cnorxz.h"
#include "raindex.h"
namespace CNORXZ
{
@ -26,15 +27,18 @@ namespace CNORXZ
class RCArray
{
public:
typedef RAIndex const_iterator;
typedef RAIndex<T> const_iterator;
DEFAULT_MEMBERS(RCArray);
DEFAULT_C(RCArray);
DEFAULT_MOVE(RCArray);
RCArray(const RCArray& a);
RCArray& operator=(const RCArray& a);
/** Construct from local array object.
@param a Local array.
@param geom Rank geometry.
*/
RCArray(const Sptr<CArrayBase<T>> a, const RangePtr& geom);
RCArray(const CArrayBase<T>& a, const RangePtr& geom);
/** @copydoc CArrayBase::operator[] */
template <typename I, typename M>

7
src/opt/mpi/tests/CMakeLists.txt
View file

@ -1,8 +1,15 @@
add_definitions(-DTEST_NUMBER_FILE="${CMAKE_SOURCE_DIR}/src/tests/numbers.txt")
include_directories(${CMAKE_SOURCE_DIR}/src/tests)
add_executable(mpirrutest rrange_unit_test.cc)
add_dependencies(mpirrutest cnorxz cnorxzmpi test_lib)
target_link_libraries(mpirrutest ${GTEST_BOTH_LIBRARIES} ${CMAKE_THREAD_LIBS_INIT} ${MPI_LIBS} cnorxz cnorxzmpi test_lib)
set(MPI_TEST_COMMAND mpirun -n 4 mpirrutest)
add_test(NAME mpirrutest COMMAND ${MPI_TEST_COMMAND})
add_executable(mpirautest rarray_unit_test.cc)
add_dependencies(mpirautest cnorxz cnorxzmpi test_lib)
target_link_libraries(mpirautest ${GTEST_BOTH_LIBRARIES} ${CMAKE_THREAD_LIBS_INIT} ${MPI_LIBS} cnorxz cnorxzmpi test_lib)
set(MPI_TEST_COMMAND mpirun -n 4 mpirautest)
add_test(NAME mpirautest COMMAND ${MPI_TEST_COMMAND})

34
src/opt/mpi/tests/rarray_unit_test.cc
View file

@ -77,16 +77,23 @@ namespace
mGeom = YRangeFactory(gs).create();
mRRange = rrange(mGRange, mGeom);
const SizeT size = ts.size()*xs.size()*xs.size()*xs.size();
Vector<Double> vec = Numbers::get(0,size/4+10);
Vector<Double> data(size);
Vector<Double> mData(size*4);
const SizeT locsize = size/4;
Vector<Double> vec = Numbers::get(0,locsize+10);
Vector<Double> data(locsize);
mData.resize(size);
const SizeT myrank = getRankNumber();
for(SizeT i = 0; i != size; ++i){
for(SizeT i = 0; i != locsize; ++i){
assert(i < data.size());
data[i] = vec[i] * vec[i+myrank] / vec[i+2*myrank];
mData[i + size*myrank] = data[i];
MPI_Bcast(mData.data() + size*i, size, MPI_DOUBLE, i, MPI_COMM_WORLD);
assert(i + locsize*myrank < mData.size());
mData[i + locsize*myrank] = data[i];
}
mLoc = std::make_shared<MArray<Double>>( mRRange->sub(1), data);
MPI_Barrier(MPI_COMM_WORLD);
for(SizeT r = 0; r != 4; ++r){
MPI_Bcast(mData.data() + locsize*r, locsize, MPI_DOUBLE, r, MPI_COMM_WORLD);
MPI_Barrier(MPI_COMM_WORLD);
}
mLoc = MArray<Double>( mRRange->sub(1), data);
mA = RCArray<Double>(mLoc, mGeom);
}
@ -95,22 +102,23 @@ namespace
RangePtr mGRange;
RangePtr mGeom;
RangePtr mRRange;
Sptr<MArray<Double>> mLoc;
RCArray mA;
MArray<Double> mLoc;
RCArray<Double> mA;
Vector<Double> mData;
};
TEST_F(RArray_Test, Basics)
TEST_F(RCArray_Test, Basics)
{
EXPECT_EQ(mA.size(), mRRange->size());
}
TEST_F(RArray_Test, GlobalIterate)
TEST_F(RCArray_Test, GlobalIterate)
{
const SizeT size = mRRange->sub(1)->size();
auto e = mA.end();
for(auto i = mA.begin(); i != e; ++i){
EXPECT_EQ(*i, mData[i.rank()*size + i.local().pos()]);
const Double x = *i;
EXPECT_EQ(x, mData[i.rank()*size + i.local()->pos()]);
}
}
}