HeterogeneousCore/MPICore/interface/MPIToken.h - master - githubmirror/cmssw

HeterogeneousCore/MPICore/interface/MPIToken.h

26 строк635 B

Andrea Bocci

Redesign the MPI event and stream synchronisation

27 апр 2026, 15:05

27 апр 2026, 15:057eadb80

#ifndef HeterogeneousCore_MPICore_interface_MPIToken_h
#define HeterogeneousCore_MPICore_interface_MPIToken_h

// C++ standard library headers
#include <memory>

// forward declaration
class MPIChannel;

class MPIToken {
public:
  // default constructor, needed to write the type's dictionary
  MPIToken() = default;

  // user-defined constructor
  explicit MPIToken(MPIChannel& channel);

  // access the data member
  MPIChannel* channel() const { return channel_.get(); }

private:
  // wrap the MPI communicator and destination
  std::shared_ptr<MPIChannel> channel_;
};

#endif  // HeterogeneousCore_MPICore_interface_MPIToken_h