portelli/Grid
1
0
Fork 0
mirror of https://github.com/paboyle/Grid.git synced 2025-04-09 21:50:45 +01:00
Code Releases Activity

Hadrons: meson field chunked HDF5 IO

Browse Source
This commit is contained in:
Antonin Portelli 2018-08-14 16:40:29 +01:00
parent cc2780bea3
commit dc6d8686de
1 changed files with 16 additions and 9 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

25
extras/Hadrons/Modules/MContraction/A2AMesonField.hpp
View File

@ -374,8 +374,10 @@ void TA2AMesonField<FImpl>::execute(void)
stopTimer("IO: total"); stopTimer("IO: total");
blockSize = static_cast<double>(nmom*ngamma*nt*N_ii*N_jj*sizeof(Complex)); blockSize = static_cast<double>(nmom*ngamma*nt*N_ii*N_jj*sizeof(Complex));
ioTime += getDTimer("IO: write block"); ioTime += getDTimer("IO: write block");
LOG(Message) << "HDF5 IO done " << blockSize/ioTime*1.0e6/1024/1024 LOG(Message) << "HDF5 IO done " << sizeString(blockSize) << " in "
<< " MB/s" << std::endl; << getTimer("IO: write block") << " us ("
<< blockSize/ioTime*1.0e6/1024/1024
<< " MB/s)" << std::endl;
} }
} }
@ -420,19 +422,24 @@ void TA2AMesonField<FImpl>::initFile(unsigned int m, unsigned int g)
int N_i = w.size(); int N_i = w.size();
int N_j = v.size(); int N_j = v.size();
Hdf5Writer writer(f); Hdf5Writer writer(f);
std::vector<hsize_t> dim = {static_cast<hsize_t>(nt), std::vector<hsize_t> dim = {static_cast<hsize_t>(nt),
static_cast<hsize_t>(N_i), static_cast<hsize_t>(N_i),
static_cast<hsize_t>(N_j)}; static_cast<hsize_t>(N_j)},
H5NS::DataSpace dataspace(dim.size(), dim.data()); chunk = {static_cast<hsize_t>(nt),
H5NS::DataSet dataset; static_cast<hsize_t>(par().block),
static_cast<hsize_t>(par().block)};
H5NS::DataSpace dataspace(dim.size(), dim.data());
H5NS::DataSet dataset;
H5NS::DSetCreatPropList plist;
push(writer, ioname(m, g)); push(writer, ioname(m, g));
write(writer, "momentum", mom_[m]); write(writer, "momentum", mom_[m]);
write(writer, "gamma", gamma_[g]); write(writer, "gamma", gamma_[g]);
auto &group = writer.getGroup(); auto &group = writer.getGroup();
plist.setChunk(chunk.size(), chunk.data());
dataset = group.createDataSet("mesonField", Hdf5Type<Complex>::type(), dataset = group.createDataSet("mesonField", Hdf5Type<Complex>::type(),
dataspace); dataspace, plist);
#else #else
HADRONS_ERROR(Implementation, "meson field I/O needs HDF5 library"); HADRONS_ERROR(Implementation, "meson field I/O needs HDF5 library");
#endif #endif