pyShamcomm.cpp File Reference

#include "shamalgs/collective/gather_str.hpp"
#include "shamalgs/collective/reduction.hpp"
#include "shambindings/pybind11_stl.hpp"
#include "shambindings/pybindaliases.hpp"
#include "shambindings/pytypealias.hpp"
#include "shamcomm/logs.hpp"
#include "shamcomm/wrapper.hpp"
#include <pybind11/pytypes.h>
#include <unordered_map>
#include <utility>
#include <vector>

Include dependency graph for pyShamcomm.cpp:

Go to the source code of this file.

Functions
shamcomm_module	def ("get_timer", [](std::string name) { return shamcomm::mpi::get_timer(std::move(name));})
shamcomm_module	def ("get_timers", []() { return shamcomm::mpi::get_timers();})
shamcomm_module	def ("mpi_timers_delta", [](std::unordered_map< std::string, f64 > start, std::unordered_map< std::string, f64 > end) { std::unordered_map< std::string, f64 > deltas{};for(auto &k :shamcomm::mpi::get_possible_keys()) { deltas[k]=shamalgs::collective::allreduce_max(end[k] - start[k]);} return deltas;})

Variables
	ON_PYTHON_INIT

Detailed Description

Author

Timothée David–Cléris (tim.s.nosp@m.hamr.nosp@m.ock@p.nosp@m.roto.nosp@m.n.me)

Definition in file pyShamcomm.cpp.

Variable Documentation

ON_PYTHON_INIT

ON_PYTHON_INIT

Initial value:

{
 
    py::module shamcomm_module = root_module.def_submodule("comm", "comm library")

Definition at line 28 of file pyShamcomm.cpp.