scnc
/
intel-qs
mirror of https://github.com/intel/intel-qs.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Add a few methods from qhipster::mpi::Environment to Python bindings

This commit is contained in:
giangiac 2020-11-09 12:35:38 -08:00
parent c58555f267
commit f03abdeb15
1 changed files with 16 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

19
pybind11/intelqs_py.cpp
View File

@ -77,6 +77,13 @@ PYBIND11_MODULE(intelqs_py, m)
.def("UniformRandomNumbers", &qhipster::RandomNumberGenerator<double>::UniformRandomNumbers)
.def("GaussianRandomNumbers", &qhipster::RandomNumberGenerator<double>::GaussianRandomNumbers)
.def("RandomIntegersInRange", &qhipster::RandomNumberGenerator<double>::RandomIntegersInRange)
.def("GetUniformRandomNumbers",
[](qhipster::RandomNumberGenerator<double> &rng, std::size_t size,
double a, double b, std::string shared) {
std::vector<double> random_values(size);
rng.UniformRandomNumbers(random_values.data(), size, a, b, shared);
return random_values;
}, "Return an array of 'size' random number from the uniform distribution [a,b[.")
#ifdef WITH_MPI_AND_MKL
.def("SetRndStreamPtrs", &qhipster::RandomNumberGenerator<double>::SetRndStreamPtrs)
#endif
@ -264,19 +271,25 @@ PYBIND11_MODULE(intelqs_py, m)
//////////////////////////////////////////////////////////////////////////////
py::class_<Environment>(m, "MPIEnvironment")
.def(py::init<>())
.def_static("GetRank", &Environment::GetRank)
.def_static("IsUsefulRank", &Environment::IsUsefulRank)
.def_static("GetPoolRank", &Environment::GetPoolRank)
.def_static("GetStateRank", &Environment::GetStateRank)
.def_static("GetPoolSize", &Environment::GetPoolSize)
.def_static("GetStateSize", &Environment::GetStateSize)
.def_static("GetNumRanksPerNode", &Environment::GetNumRanksPerNode)
.def_static("GetNumNodes", &Environment::GetNumNodes)
.def_static("GetStateId", &Environment::GetStateId)
.def_static("GetNumStates", &Environment::GetNumStates);
.def_static("GetNumStates", &Environment::GetNumStates)
.def_static("Barrier", &qhipster::mpi::Barrier)
.def_static("PoolBarrier", &qhipster::mpi::PoolBarrier)
.def_static("StateBarrier", &qhipster::mpi::StateBarrier)
.def_static("IncoherentSumOverAllStatesOfPool", &Environment::IncoherentSumOverAllStatesOfPool<double>)
.def_static("UpdateStateComm", &Environment::UpdateStateComm);
}