Python/Communication/Comm.cpp File Reference

#include <LibUtilities/BasicUtils/SharedArray.hpp>
#include <LibUtilities/Communication/Comm.h>
#include <LibUtilities/Python/BasicUtils/SharedArray.hpp>
#include <LibUtilities/Python/NekPyConfig.hpp>
#include <vector>

Go to the source code of this file.

Functions
template<typename T >
T	AllReduce (CommSharedPtr &comm, T toReduce, ReduceOperator oper)
void	export_Comm (py::module &m)
	Export for Comm communicator. More...

Function Documentation

AllReduce()

template<typename T >

T AllReduce	(	CommSharedPtr &	comm,
		T	toReduce,
		ReduceOperator	oper
	)

Definition at line 47 of file Python/Communication/Comm.cpp.

{
    comm->AllReduce(toReduce, oper);
    return toReduce;
}

Referenced by export_Comm().

export_Comm()

void export_Comm

(

py::module &

m

)

Export for Comm communicator.

Definition at line 56 of file Python/Communication/Comm.cpp.

{
    // Export ReduceOperator enum
    NEKPY_WRAP_ENUM(m, ReduceOperator, ReduceOperatorMap);
 
    py::class_<Comm, std::shared_ptr<Comm>>(m, "Comm")
        .def("GetSize", &Comm::GetSize)
        .def("GetRank", &Comm::GetRank)
        .def("GetType", &Comm::GetType, py::return_value_policy::reference)
        .def("AllReduce", &AllReduce<double>)
        .def("AllReduce", &AllReduce<int>)
        .def("AllReduce", &AllReduce<long>)
        .def("AllReduce", &AllReduce<Array<OneD, NekDouble>>);
}

References AllReduce(), NEKPY_WRAP_ENUM, and Nektar::LibUtilities::ReduceOperatorMap.

Referenced by PYBIND11_MODULE().