enable MT

fastmachinelearning · Sep 18, 2018 · 6f1fe6d · 6f1fe6d
1 parent 9b847af
commit 6f1fe6d
Showing 1 changed file with 8 additions and 0 deletions.
diff --git a/AnalysisFW/python/jetImageTest_mc_cfg.py b/AnalysisFW/python/jetImageTest_mc_cfg.py
@@ -9,6 +9,8 @@
 options.register("port", -1, VarParsing.multiplicity.singleton, VarParsing.varType.int)
 options.register("timeout", 30, VarParsing.multiplicity.singleton, VarParsing.varType.int)
 options.register("params", "", VarParsing.multiplicity.singleton, VarParsing.varType.string)
+options.register("threads", 1, VarParsing.multiplicity.singleton, VarParsing.varType.int)
+options.register("streams", 0, VarParsing.multiplicity.singleton, VarParsing.varType.int)
 options.parseArguments()
 
 if len(options.params)>0 and options.remote:
@@ -74,3 +76,9 @@
             limit = cms.untracked.int32(10000000),
         )
     )
+
+if options.threads>0:
+    if not hasattr(process,"options"):
+        process.options = cms.untracked.PSet()
+    process.options.numberOfThreads = cms.untracked.uint32(options.threads)
+    process.options.numberOfStreams = cms.untracked.uint32(options.streams if options.streams>0 else 0)