From 1af1c018cb5c618c31a242ce2954e6dd41bcf151 Mon Sep 17 00:00:00 2001 From: jsamudio Date: Thu, 26 Oct 2023 13:45:37 -0500 Subject: [PATCH 1/2] Add Alpaka Process Modifier --- Configuration/ProcessModifiers/python/alpaka_cff.py | 5 +++++ Configuration/StandardSequences/python/Services_cff.py | 8 +++++++- 2 files changed, 12 insertions(+), 1 deletion(-) create mode 100644 Configuration/ProcessModifiers/python/alpaka_cff.py diff --git a/Configuration/ProcessModifiers/python/alpaka_cff.py b/Configuration/ProcessModifiers/python/alpaka_cff.py new file mode 100644 index 0000000000000..8073154217e61 --- /dev/null +++ b/Configuration/ProcessModifiers/python/alpaka_cff.py @@ -0,0 +1,5 @@ +import FWCore.ParameterSet.Config as cms + +# This modifier is for enabling alpaka modules + +alpaka = cms.Modifier() diff --git a/Configuration/StandardSequences/python/Services_cff.py b/Configuration/StandardSequences/python/Services_cff.py index c8c0b1e63b226..fa57269be686c 100644 --- a/Configuration/StandardSequences/python/Services_cff.py +++ b/Configuration/StandardSequences/python/Services_cff.py @@ -13,9 +13,15 @@ def _addProcessAccelerators(process): process.load("Configuration.StandardSequences.Accelerators_cff") +def _addProcessAcceleratorsAlpaka(process): + process.load("HeterogeneousCore.AlpakaCore.ProcessAcceleratorAlpaka_cfi") + from Configuration.ProcessModifiers.gpu_cff import gpu from Configuration.ProcessModifiers.pixelNtupletFit_cff import pixelNtupletFit -modifyConfigurationStandardSequencesServicesAddProcessAccelerators_ = (gpu | pixelNtupletFit).makeProcessModifier(_addProcessAccelerators) +from Configuration.ProcessModifiers.alpaka_cff import alpaka + +modifyConfigurationStandardSequencesServicesAddProcessAccelerators_ = (gpu | pixelNtupletFit | alpaka).makeProcessModifier(_addProcessAccelerators) +modifyConfigurationStandardSequencesServicesAddProcessAcceleratorsAlpaka_ = alpaka.makeProcessModifier(_addProcessAcceleratorsAlpaka) # load TritonService when SONIC workflow is enabled def _addTritonService(process): From 11e604f781867e16a6c0494aba1c5472ca2de054 Mon Sep 17 00:00:00 2001 From: jsamudio Date: Fri, 10 Nov 2023 10:46:43 -0600 Subject: [PATCH 2/2] Load alpaka process accelerator with other accelerators --- Configuration/StandardSequences/python/Accelerators_cff.py | 1 + Configuration/StandardSequences/python/Services_cff.py | 4 ---- 2 files changed, 1 insertion(+), 4 deletions(-) diff --git a/Configuration/StandardSequences/python/Accelerators_cff.py b/Configuration/StandardSequences/python/Accelerators_cff.py index f313750526119..764a80abbd772 100644 --- a/Configuration/StandardSequences/python/Accelerators_cff.py +++ b/Configuration/StandardSequences/python/Accelerators_cff.py @@ -5,3 +5,4 @@ from HeterogeneousCore.CUDACore.ProcessAcceleratorCUDA_cfi import ProcessAcceleratorCUDA from HeterogeneousCore.ROCmCore.ProcessAcceleratorROCm_cfi import ProcessAcceleratorROCm +from HeterogeneousCore.AlpakaCore.ProcessAcceleratorAlpaka_cfi import ProcessAcceleratorAlpaka diff --git a/Configuration/StandardSequences/python/Services_cff.py b/Configuration/StandardSequences/python/Services_cff.py index fa57269be686c..d7530b51f0ed9 100644 --- a/Configuration/StandardSequences/python/Services_cff.py +++ b/Configuration/StandardSequences/python/Services_cff.py @@ -13,15 +13,11 @@ def _addProcessAccelerators(process): process.load("Configuration.StandardSequences.Accelerators_cff") -def _addProcessAcceleratorsAlpaka(process): - process.load("HeterogeneousCore.AlpakaCore.ProcessAcceleratorAlpaka_cfi") - from Configuration.ProcessModifiers.gpu_cff import gpu from Configuration.ProcessModifiers.pixelNtupletFit_cff import pixelNtupletFit from Configuration.ProcessModifiers.alpaka_cff import alpaka modifyConfigurationStandardSequencesServicesAddProcessAccelerators_ = (gpu | pixelNtupletFit | alpaka).makeProcessModifier(_addProcessAccelerators) -modifyConfigurationStandardSequencesServicesAddProcessAcceleratorsAlpaka_ = alpaka.makeProcessModifier(_addProcessAcceleratorsAlpaka) # load TritonService when SONIC workflow is enabled def _addTritonService(process):