Merge pull request cms-sw#9 from kpedro88/TritonCompression

propagate client interface updates
yongbinfeng · Jul 30, 2021 · b2a9785 · b2a9785
2 parents 39e5f5f + 1a91c7f
commit b2a9785
Show file tree

Hide file tree

Showing 4 changed files with 12 additions and 4 deletions.
diff --git a/RecoBTag/ONNXRuntime/python/pfParticleNetAK4_cff.py b/RecoBTag/ONNXRuntime/python/pfParticleNetAK4_cff.py
@@ -25,12 +25,14 @@
     preprocess_json = 'RecoBTag/Combined/data/ParticleNetAK4/CHS/V00/preprocess_noragged.json',
     Client = cms.PSet(
         timeout = cms.untracked.uint32(300),
-        modelName = cms.string("particlenet_AK4"),
         mode = cms.string("Async"),
+        modelName = cms.string("particlenet_AK4"),
         modelConfigPath = cms.FileInPath("HeterogeneousCore/SonicTriton/data/models/particlenet_AK4/config.pbtxt"),
         modelVersion = cms.string(""),
         verbose = cms.untracked.bool(False),
         allowedTries = cms.untracked.uint32(0),
+        useSharedMemory = cms.untracked.bool(True),
+        compression = cms.untracked.string(""),
     ),
     flav_names = pfParticleNetAK4JetTags.flav_names,
 ))

diff --git a/RecoBTag/ONNXRuntime/python/pfParticleNet_cff.py b/RecoBTag/ONNXRuntime/python/pfParticleNet_cff.py
@@ -25,12 +25,14 @@
     preprocess_json = 'RecoBTag/Combined/data/ParticleNetAK8/General/V01/preprocess_noragged.json',
     Client = cms.PSet(
         timeout = cms.untracked.uint32(300),
-        modelName = cms.string("particlenet"),
         mode = cms.string("Async"),
+        modelName = cms.string("particlenet"),
         modelConfigPath = cms.FileInPath("HeterogeneousCore/SonicTriton/data/models/particlenet/config.pbtxt"),
         modelVersion = cms.string(""),
         verbose = cms.untracked.bool(False),
         allowedTries = cms.untracked.uint32(0),
+        useSharedMemory = cms.untracked.bool(True),
+        compression = cms.untracked.string(""),
     ),
     flav_names = pfParticleNetJetTags.flav_names,
 ))

diff --git a/RecoMET/METPUSubtraction/python/deepMETSonicProducer_cff.py b/RecoMET/METPUSubtraction/python/deepMETSonicProducer_cff.py
@@ -3,14 +3,16 @@
 deepMETSonicProducer = cms.EDProducer("DeepMETSonicProducer",
     Client = cms.PSet(
         timeout = cms.untracked.uint32(300),
-        modelName = cms.string("deepmet"),
         mode = cms.string("Async"),
+        modelName = cms.string("deepmet"),
         modelConfigPath = cms.FileInPath("HeterogeneousCore/SonicTriton/data/models/deepmet/config.pbtxt"),
         # version "1" is the resolutionTune
         # version "2" is the responeTune
         modelVersion = cms.string("1"),
         verbose = cms.untracked.bool(False),
         allowedTries = cms.untracked.uint32(0),
+        useSharedMemory = cms.untracked.bool(True),
+        compression = cms.untracked.string(""),
     ),
     pf_src = cms.InputTag("packedPFCandidates"),
 )
diff --git a/RecoTauTag/RecoTau/python/deepTauSonicProducer_cff.py b/RecoTauTag/RecoTau/python/deepTauSonicProducer_cff.py
@@ -3,12 +3,14 @@
 deepTauSonicProducer = cms.EDProducer("DeepTauIdSonicProducer",
     Client = cms.PSet(
         timeout = cms.untracked.uint32(300),
-        modelName = cms.string("deeptau_ensemble"),
         mode = cms.string("Async"),
+        modelName = cms.string("deeptau_ensemble"),
         modelConfigPath = cms.FileInPath("HeterogeneousCore/SonicTriton/data/models/deeptau_ensemble/config.pbtxt"),
         modelVersion = cms.string(""),
         verbose = cms.untracked.bool(False),
         allowedTries = cms.untracked.uint32(0),
+        useSharedMemory = cms.untracked.bool(True),
+        compression = cms.untracked.string(""),
     ),
     electrons = cms.InputTag('slimmedElectrons'),
     muons = cms.InputTag('slimmedMuons'),