HeterogeneousCore/MPICore/test/controller_cfg.py - master - githubmirror/cmssw

HeterogeneousCore/MPICore/test/controller_cfg.py

65 строк2 KB

Andrea Bocci

Replace LogAbsolute with LogInfo

25 апр 2026, 20:25

25 апр 2026, 20:25c23f417

import FWCore.ParameterSet.Config as cms

process = cms.Process("MPIController")

process.options.numberOfThreads = 4
process.options.numberOfStreams = 4
# MPIController supports a single concurrent LuminosityBlock
process.options.numberOfConcurrentLuminosityBlocks = 1
process.options.numberOfConcurrentRuns = 1
process.options.wantSummary = False

process.load("FWCore.ParameterSet.MessageLogger")
process.MessageLogger.cerr.MPI = cms.untracked.PSet(
    reportEvery = cms.untracked.int32( 1 ),
    limit = cms.untracked.int32( 10000000 )
)

process.load("HeterogeneousCore.MPIServices.MPIService_cfi")

from eventlist_cff import eventlist
process.source = cms.Source("EmptySourceFromEventIDs",
    events = cms.untracked(eventlist)
)

process.maxEvents.input = 100

from HeterogeneousCore.MPICore.modules import *

process.mpiController = MPIController(
    mode = 'CommWorld',
    followerProcessName = 'Follower'
)

process.ids = cms.EDProducer("edmtest::EventIDProducer")

process.initialcheck = cms.EDAnalyzer("edmtest::EventIDValidator",
    source = cms.untracked.InputTag('ids')
)

process.sender = MPISender(
    upstream = "mpiController",
    instance = 42,
    products = [ "edmEventID_ids__*" ]
)

process.othersender = MPISender(
    upstream = "mpiController",
    instance = 19,
    products = [ "edmEventID_ids__*" ]
)

process.receiver = MPIReceiver(
    upstream = "othersender",   # guarantees that this module will only run after "othersender" has run
    instance = 99,
    products = [ dict(
        type = "edm::EventID",
        label = ""
    )]
)

process.finalcheck = cms.EDAnalyzer("edmtest::EventIDValidator",
    source = cms.untracked.InputTag('receiver')
)

process.path = cms.Path(process.mpiController + process.ids + process.initialcheck + process.sender + process.othersender + process.receiver + process.finalcheck)