Merge pull request #11 from hatakeyamak/PFRecHitAndCluster_GPU_12_5_h…

…ackason2 Updates (kernel optimizations, cleanup, etc)
jsamudio · Oct 5, 2022 · ac400fc · ac400fc
2 parents a71d4e8 + af2e681
commit ac400fc
Show file tree

Hide file tree

Showing 6 changed files with 187 additions and 142 deletions.
diff --git a/RecoLocalCalo/Configuration/python/customizeHcalOnlyForProfiling.py b/RecoLocalCalo/Configuration/python/customizeHcalOnlyForProfiling.py
@@ -23,7 +23,7 @@ def customizeHcalOnlyForProfilingGPUOnly(process):
 # Currently, this means:
 #   - running the unpacker on CPU, converting the digis into SoA format and copying them to GPU
 #   - running the HBHE local reconstruction, including MAHI, on GPU.
-#   - running the HBHE PFRecHit and PFCluster producers on GPU [including copy of them to host, as we cannot untangle the copy back at a moment]
+#   - running the HBHE PFRecHit and PFCluster producers on GPU [without copy of them to host]
 def customizeHcalPFOnlyForProfilingGPUOnly(process):
 
   process.consumer = cms.EDAnalyzer("GenericConsumer",
@@ -33,6 +33,8 @@ def customizeHcalPFOnlyForProfilingGPUOnly(process):
   process.consume_step = cms.EndPath(process.consumer)
 
   process.schedule = cms.Schedule(process.raw2digi_step, process.reconstruction_step, process.consume_step)
+  process.particleFlowClusterHBHEOnly.cuda.produceLegacy = cms.bool(False)
+  process.particleFlowRecHitHBHEOnly.cuda.produceLegacy = cms.bool(False)
 
   return process
 

diff --git a/RecoParticleFlow/PFClusterProducer/plugins/DeclsForKernels.h b/RecoParticleFlow/PFClusterProducer/plugins/DeclsForKernels.h
@@ -161,8 +161,8 @@ namespace PFRecHit {
 namespace PFClustering {
   namespace HCAL {
     struct ConfigurationParameters {
-      uint32_t maxRH = 3000;          // previously: 2000
-      uint32_t maxPFCFracs = 600000;  // previously: 80000
+      uint32_t maxRH = 4000;          // previously: 2000
+      uint32_t maxPFCFracs = 200000;  // previously: 80000
       uint32_t maxNeighbors = 8;
     };
 
@@ -332,6 +332,7 @@ namespace PFClustering {
       }
     };
   }  // namespace HCAL
+
 }  // namespace PFClustering
 
 #endif  // RecoParticleFlow_PFClusterProducer_plugins_DeclsForKernels_h