Add cmsDriver option --maxmem_profile to prefix LD_PRELOAD="libPerfToolsAllocMonitorPreload.so libPerfToolsMaxMemoryPreload.so" to cmsRun

Configuration/Applications/python/Options.py

@@ -428,6 +428,12 @@
                             type=str,
                             dest="heap_profile")
 
+expertSettings.add_argument("--maxmem_profile",
+                            help="add the PerfTools/MaxMemoryPreload Service",
+                            default=None,
+                            type=str,
+                            dest="maxmem_profile")
+
 expertSettings.add_argument("--io",
                             help="Create a json file with io informations",
                             default=None,

Configuration/Applications/python/cmsDriverOptions.py

@@ -17,7 +17,7 @@ def checkModifier(era):
 
 def checkOptions():
     return
-    
+
 def adaptOptions():
     return
 
@@ -34,7 +34,7 @@ def OptionsFromCommand(command):
 def OptionsFromCommandLine():
     import sys
     options=OptionsFromItems(sys.argv[1:])
-    # memorize the command line arguments 
+    # memorize the command line arguments
     options.arguments = reduce(lambda x, y: x+' '+y, sys.argv[1:])
     return options
 
@@ -60,8 +60,8 @@ def OptionsFromItems(items):
     if options.triggerResultsProcess == None and "ALCAOUTPUT" in options.step:
         print("ERROR: If ALCA splitting is requested, the name of the process in which the alca producers ran needs to be specified. E.g. via --triggerResultsProcess RECO")
         sys.exit(1)
-            
-    if not options.evt_type:            
+
+    if not options.evt_type:
         options.evt_type=sys.argv[1]
 
     #now adjust the given parameters before passing it to the ConfigBuilder
@@ -123,15 +123,15 @@ def OptionsFromItems(items):
 
     # add on the end of job sequence...
     addEndJob = True
-    if ("FASTSIM" in options.step and not "VALIDATION" in options.step) or "HARVESTING" in options.step or "ALCAHARVEST" in options.step or "ALCAOUTPUT" in options.step or options.step == "": 
+    if ("FASTSIM" in options.step and not "VALIDATION" in options.step) or "HARVESTING" in options.step or "ALCAHARVEST" in options.step or "ALCAOUTPUT" in options.step or options.step == "":
         addEndJob = False
     if ("SKIM" in options.step and not "RECO" in options.step):
         addEndJob = False
     if ("ENDJOB" in options.step):
         addEndJob = False
     if ('DQMIO' in options.datatier):
         addEndJob = False
-    if addEndJob:    
+    if addEndJob:
         options.step=options.step+',ENDJOB'
 
 
@@ -180,7 +180,7 @@ def OptionsFromItems(items):
             options.name = 'RESIM'
         elif 'reDIGI' in options.trimmedStep:
             options.name = 'REDIGI'
-        elif 'HLT' in options.trimmedStep:    
+        elif 'HLT' in options.trimmedStep:
             options.name = 'HLT'
         elif 'RECO' in options.trimmedStep:
             options.name = 'RECO'
@@ -244,13 +244,19 @@ def OptionsFromItems(items):
             raise Exception("Not a valid profiler type %s. Alternatives are pp, mp, fp=<function>."%(profilerType))
 
         options.prefix = "igprof -t cmsRun -%s" % profilerType
-        
+
     if options.heap_profile:
-        if options.profile and options.prefix:
+        if options.heap_profile and options.prefix:
             raise Exception("--heap_profile and --prefix are incompatible")
         profilerType = 'pp'
         options.prefix = "MALLOC_CONF=prof:true,prof_accum:true,prof_prefix:jeprof.out cmsRunJE "
 
+    if options.maxmem_profile:
+        if options.maxmem_profile and and options.prefix:
+            raise Exception("--maxmem_profile and --prefix are incompatible")
+        options.prefix = 'LD_PRELOAD="libPerfToolsAllocMonitorPreload.so libPerfToolsMaxMemoryPreload.so" cmsRun '
+
+
     # If an "era" argument was supplied make sure it is one of the valid possibilities
     if options.era :
         from Configuration.StandardSequences.Eras import eras
@@ -262,7 +268,7 @@ def OptionsFromItems(items):
                 validOptions="" # Create a stringified list of valid options to print to the user
                 for key in eras.__dict__ :
                     if checkModifier(eras.__dict__[key]):
-                        if validOptions!="" : validOptions+=", " 
+                        if validOptions!="" : validOptions+=", "
                         validOptions+="'"+key+"'"
                 raise Exception( "'%s' is not a valid option for '--era'. Valid options are %s." % (eraName, validOptions) )
     # If the "--fast" option was supplied automatically enable the fastSim era