From 7123df376654c2a5fc19591207b036a6c5c47119 Mon Sep 17 00:00:00 2001 From: Andreas Psallidas Date: Sun, 24 Jan 2021 15:58:40 +0100 Subject: [PATCH] use latest scram and gpu support --- SubmitFileGSD.sh | 2 +- SubmitHGCalPGun.py | 2 ++ 2 files changed, 3 insertions(+), 1 deletion(-) diff --git a/SubmitFileGSD.sh b/SubmitFileGSD.sh index 019b30d..20ea5a6 100755 --- a/SubmitFileGSD.sh +++ b/SubmitFileGSD.sh @@ -19,7 +19,7 @@ keepDQMfile=${14} ##Create Work Area export SCRAM_ARCH=${CMSSWARCH} -source /afs/cern.ch/cms/cmsset_default.sh +source /cvmfs/cms.cern.ch/cmsset_default.sh eval `scramv1 project CMSSW ${CMSSWVER}` cd ${CMSSWVER}/ rm -rf ./* diff --git a/SubmitHGCalPGun.py b/SubmitHGCalPGun.py index 498b6c3..935e7ac 100644 --- a/SubmitHGCalPGun.py +++ b/SubmitHGCalPGun.py @@ -56,6 +56,7 @@ def createParser(): parser.add_option('', '--storePFCandidates', action='store_true', dest='storePFCandidates', default=False, help='store PFCandidates collection') parser.add_option('', '--multiClusterTag', action='store', dest='MULTICLUSTAG', default="hgcalMultiClusters", help='name of HGCalMultiCluster InputTag - use hgcalLayerClusters before CMSSW_10_3_X') parser.add_option('', '--keepDQMfile', action='store_true', dest='DQM', default=False, help='store the DQM file in relevant folder locally or in EOS, default is False.') + parser.add_option('', '--requestGPUs', action='store_true', dest='GPU', default=False, help='if used then it is set to True and will use in condor GPU machines, default is False. Keep in mind that GPU machines are limited in contrary to CPU.') return parser @@ -483,6 +484,7 @@ def submitHGCalProduction(*args, **kwargs): write_condorjob.write('output = '+outDir+'/std/'+basename+'.out \n') write_condorjob.write('error = '+outDir+'/std/'+basename+'.err \n') write_condorjob.write('log = '+outDir+'/std/'+basename+'_htc.log \n\n') + if (opt.GPU): write_condorjob.write('request_GPUs = 1\n') write_condorjob.write('max_retries = 1\n') write_condorjob.write('queue \n') write_condorjob.close()