From 6f1fe6d4c7c88141f3b42c46767b568e1e5ef158 Mon Sep 17 00:00:00 2001 From: Kevin Pedro Date: Tue, 18 Sep 2018 11:15:43 -0500 Subject: [PATCH] enable MT --- AnalysisFW/python/jetImageTest_mc_cfg.py | 8 ++++++++ 1 file changed, 8 insertions(+) diff --git a/AnalysisFW/python/jetImageTest_mc_cfg.py b/AnalysisFW/python/jetImageTest_mc_cfg.py index 9004158..dd02c80 100644 --- 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)