Validation/Performance/python/TimeMemoryInfo.py - master - githubmirror/cmssw

Validation/Performance/python/TimeMemoryInfo.py

23 строки846 B

Matti Kortelainen

Add customiseWithTimeMemoryInfo() function

05 дек 2024, 20:11

05 дек 2024, 20:11fa6807f

import FWCore.ParameterSet.Config as cms
def customise(process):

    #Adding SimpleMemoryCheck service:
    process.SimpleMemoryCheck=cms.Service("SimpleMemoryCheck",
                                          ignoreTotal=cms.untracked.int32(1),
                                          oncePerEventMode=cms.untracked.bool(True))
    #Adding Timing service:
    process.Timing=cms.Service("Timing")
    
    #Add these 3 lines to put back the summary for timing information at the end of the logfile
    #(needed for TimeReport report)
    if hasattr(process,'options'):
        process.options.wantSummary = cms.untracked.bool(True)
    else:
        process.options = cms.untracked.PSet(
            wantSummary = cms.untracked.bool(True)
        )

    return(process)

def customiseWithTimeMemoryInfo(process):
    return customise(process)