RAthExHiveOpts.py

Go to the documentation of this file.

#
#  Copyright (C) 2002-2024 CERN for the benefit of the ATLAS collaboration
#
 
 
# 
 
from AthenaCommon.Logging import log as msg
 
from AthenaCommon.ConcurrencyFlags import jobproperties as jp
nThreads = jp.ConcurrencyFlags.NumThreads()
if (nThreads < 1) :
   msg.fatal('numThreads must be >0. Did you set the --threads=N option?')
   sys.exit(AthenaCommon.ExitCodes.CONFIGURATION_ERROR)
 
#
 
 
from AthenaServices.AthenaServicesConf import AthenaHiveEventLoopMgr
# svcMgr.AthenaHiveEventLoopMgr.OutputLevel = INFO
 
from StoreGate.StoreGateConf import SG__HiveMgrSvc
# svcMgr.EventDataSvc.OutputLevel = INFO
 
from GaudiHive.GaudiHiveConf import AlgResourcePool
# svcMgr += AlgResourcePool( OutputLevel = INFO );
 
from GaudiHive.GaudiHiveConf import ForwardSchedulerSvc
svcMgr.ForwardSchedulerSvc.OutputLevel = INFO
svcMgr.ForwardSchedulerSvc.CheckDependencies = True
 
#
 
 
# numStores = 1
# numAlgsInFlight = 1
 
# svcMgr.EventDataSvc.NSlots = numStores
# svcMgr.ForwardSchedulerSvc.MaxEventsInFlight = numStores
# svcMgr.ForwardSchedulerSvc.MaxAlgosInFlight = numAlgsInFlight
 
# ThreadPoolService thread local initialization
# from GaudiHive.GaudiHiveConf import ThreadPoolSvc
# svcMgr += ThreadPoolSvc("ThreadPoolSvc")
# svcMgr.ThreadPoolSvc.ThreadInitTools = ["ThreadInitTool"]
 
#---------------------------------------------------------------------------------#
 
#
 
 
#---------------------------------------------------------------------------------#
 
 
#
 
 
from AthenaCommon.AlgSequence import AlgSequence
topSequence = AlgSequence()
 
from SGComps.SGCompsConf import SGInputLoader
topSequence+=SGInputLoader(OutputLevel=INFO, ShowEventDump=False)
topSequence.SGInputLoader.Load = { ('EventInfo','McEventInfo') }
 
from xAODEventInfoCnv.xAODEventInfoCnvConf import xAODMaker__EventInfoCnvAlg
topSequence += xAODMaker__EventInfoCnvAlg()
 
from AthExHive.AthExHiveConf import *
topSequence+=HiveAlgR(OutputLevel=DEBUG)
topSequence+=HiveAlgA(OutputLevel=DEBUG,Time=20)
topSequence+=HiveAlgB(OutputLevel=DEBUG,Time=10)
topSequence+=HiveAlgC(OutputLevel=DEBUG,Time=190)
 
#--------------------------------------------------------------
# Event related parameters
#--------------------------------------------------------------
 
theApp.EvtMax = 20
 
nProc = jp.ConcurrencyFlags.NumProcs()
if (nProc > 0) :
 
   #
   
   import AthenaCommon.AtlasUnixGeneratorJob
 
   #
   
 
   from AthenaCommon.Logging import log as msg
   if (theApp.EvtMax == -1) : 
      msg.fatal('EvtMax must be >0 for hybrid configuration')
      sys.exit(AthenaCommon.ExitCodes.CONFIGURATION_ERROR)
 
   if ( theApp.EvtMax % nProc != 0 ) :
      msg.warning('EvtMax[%s] is not divisible by nProcs[%s]: MP Workers will not process all requested events',theApp.EvtMax,nProc)
 
   chunkSize = int (theApp.EvtMax / nProc)
 
   from AthenaMP.AthenaMPFlags import jobproperties as jps 
   jps.AthenaMPFlags.PreCountedEvents=theApp.EvtMax
   jps.AthenaMPFlags.ChunkSize= chunkSize
 
   msg.info('AthenaMP workers will process %s events each',chunkSize)
 
 
 
print "==========================================================================================\n"
 
#
 
 
#  set algCardinality = 1 to disable cloning for all Algs
algCardinality = jp.ConcurrencyFlags.NumThreads()
 
if (algCardinality != 1):
   for alg in topSequence:
       name = alg.name()
       alg.Cardinality = algCardinality