IMPIClusterSvc.h

Go to the documentation of this file.

/*
  Copyright (C) 2002-2025 CERN for the benefit of the ATLAS collaboration
*/
#ifndef ATHENAKERNEL_IMPICLUSTERSVC_H
#define ATHENAKERNEL_IMPICLUSTERSVC_H
 
#include <cstdint>
#include <string>
 
#include "GaudiKernel/IInterface.h"
 
struct ClusterMessage;
 
namespace boost::mpi3 {
class communicator;
}
 
enum class ClusterComm { Default, EventData };

class IMPIClusterSvc : virtual public IInterface {
 public:
  DeclareInterfaceID(IMPIClusterSvc, 1, 0);

  virtual int numRanks() const = 0;

  virtual int rank() const = 0;

  virtual void barrier() = 0;

  virtual void abort() = 0;

  virtual void sendMessage(int destRank, ClusterMessage message,
                           ClusterComm communicator = ClusterComm::Default) = 0;

  virtual ClusterMessage waitReceiveMessage(
      ClusterComm communicator = ClusterComm::Default) = 0;

  virtual boost::mpi3::communicator& data_communicator() = 0;

  virtual void log_addEvent(int eventIdx, std::int64_t run_number,
                            std::int64_t event_number,
                            std::int64_t request_time_ns,
                            std::size_t slot) = 0;

  virtual void log_completeEvent(std::int64_t run_number,
                                 std::int64_t event_number,
                                 std::int64_t status) = 0;
};
 
#endif  // ATHENAKERNEL_IMPICLUSTERSVC_H