TracccTritonBootstrap Namespace Reference

Functions
dict	bootstrap (int nSlots=1, str sequenceName=DEFAULT_SEQUENCE_NAME)

Detailed Description

Entry point the embedded Athena application uses to configure itself.

Function Documentation

bootstrap()

dict bootstrap	(	int	nSlots = 1,
		str	sequenceName = DEFAULT_SEQUENCE_NAME )

Configure and initialize the embedded Athena application.

Raises on failure.

@param nSlots number of event-store slots to create, one per Triton model instance
@param sequenceName name of the AthSequencer holding the device chain; the
       C++ TracccTritonRunner looks this up and executes it per request
@return the StoreGate keys the C++ side needs, as configured above.
        TracccTritonInitializer copies these into its Config, so the names
        live only in TracccTritonBackendConfig and cannot drift.

Definition at line 19 of file TracccTritonBootstrap.py.

              sequenceName: str = DEFAULT_SEQUENCE_NAME) -> dict:
    """Configure and initialize the embedded Athena application.
 
    Raises on failure.
 
    @param nSlots number of event-store slots to create, one per Triton model instance
    @param sequenceName name of the AthSequencer holding the device chain; the
           C++ TracccTritonRunner looks this up and executes it per request
    @return the StoreGate keys the C++ side needs, as configured above.
            TracccTritonInitializer copies these into its Config, so the names
            live only in TracccTritonBackendConfig and cannot drift.
    """
 
    if nSlots < 1:
        raise ValueError(
            f"TracccTritonBootstrap.bootstrap: nSlots must be >= 1, got {nSlots}"
        )
 
    flags = initConfigFlags()
 
    # GPU device chain.
    flags.Device.Backend = DeviceBackend.CUDA
    flags.Acts.Device.doClusterization = True
    flags.Acts.Device.doSpacePointFormation = True
    flags.Acts.Device.doSeeding = True
    flags.Acts.Device.doTrackReconstruction = True
    flags.CUDA.Stream = CUDAStream.PerEvent
 
    # The device detector description is converted from the ACTS Gen3 tracking geometry
    flags.Acts.TrackingGeometry.UseBlueprint = True
    flags.Acts.TrackingGeometry.BuildDetrayGeometry = True
 
    # No input / no event loop needed; the Runner drives the algorithms directly
    flags.Input.Files = []
    flags.Exec.MaxEvents = 0
 
    flags.Input.isMC = True
    flags.GeoModel.AtlasVersion = "ATLAS-P2-RUN4-03-00-01"
    flags.IOVDb.GlobalTag = "COND-MC21-SDR-RUN4-06"
 
    # The same detector as the client's job is needed for conversions
    from InDetConfig.ConfigurationHelpers import OnlyTrackingPreInclude
    OnlyTrackingPreInclude(flags)
    flags.Detector.EnableHGTD = False
 
    # Flag overrides from the tritonserver environment, e.g.
    # TRACCC_TRITON_FLAGS="Acts.Device.seedingStrategy=SeedingStrategy.Gbts"
    for flag in os.environ.get("TRACCC_TRITON_FLAGS", "").split(";"):
        if flag.strip():
            flags.fillFromString(flag)
 
    flags.lock()
 
    acc = MainServicesCfg(flags)
 
    # One event-store slot per Triton model instance.
    acc.addService(
        CompFactory.SG.HiveMgrSvc("EventDataSvc", NSlots=nSlots), create=True
    )
 
    from TracccTritonBackend.TracccTritonBackendConfig import (
        TracccTritonDeviceRecoCfg,
    )
    acc.merge(TracccTritonDeviceRecoCfg(flags, sequenceName=sequenceName))
 
    # SG::getNSlots() caches the number of event slots the first time it is
    # called, and gives 1 if EventDataSvc cannot be retrieved yet. With
    # MetaDataSvc as a proxy provider, the first StoreGateSvc (initialized by
    # GeoModelSvc) creates MetaDataSvc -> OutputStreamSequencerSvc, which asks
    # before EventDataSvc exists; PerEventStreamSvc then has streams for one
    # slot only. There is no input file, so no input metadata to provide.
    pps = acc.getService("ProxyProviderSvc")
    pps.ProviderNames = [name for name in pps.ProviderNames
                         if "MetaDataSvc" not in name]
    # MetaDataSvc also created AthenaPoolCnvSvc at initialization. Created
    # lazily instead, it is created while IOVDbSvc holds the database lock and
    # registers with IncidentSvc, while another slot's clearStore() holds the
    # IncidentSvc lock waiting for the database lock: deadlock.
    acc.addService(acc.getService("AthenaPoolCnvSvc"), create=True)
 
    # Equivalent to ComponentAccumulator.run() without the the event loop
    app = acc.createApp()
    sc = app.initialize()
    if not sc.isSuccess():
        raise RuntimeError(
            "TracccTritonBootstrap.bootstrap: ApplicationMgr.initialize() failed"
        )
 
    sc = app.start()
    if not sc.isSuccess():
        raise RuntimeError(
            "TracccTritonBootstrap.bootstrap: ApplicationMgr.start() failed"
        )
 
    return dict(STOREGATE_KEYS)