Skip to content

Commit 1af1c01

Browse files
committed
Add Alpaka Process Modifier
1 parent f0bc2a8 commit 1af1c01

File tree

2 files changed

+12
-1
lines changed

2 files changed

+12
-1
lines changed
Lines changed: 5 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,5 @@
1+
import FWCore.ParameterSet.Config as cms
2+
3+
# This modifier is for enabling alpaka modules
4+
5+
alpaka = cms.Modifier()

Configuration/StandardSequences/python/Services_cff.py

Lines changed: 7 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -13,9 +13,15 @@
1313
def _addProcessAccelerators(process):
1414
process.load("Configuration.StandardSequences.Accelerators_cff")
1515

16+
def _addProcessAcceleratorsAlpaka(process):
17+
process.load("HeterogeneousCore.AlpakaCore.ProcessAcceleratorAlpaka_cfi")
18+
1619
from Configuration.ProcessModifiers.gpu_cff import gpu
1720
from Configuration.ProcessModifiers.pixelNtupletFit_cff import pixelNtupletFit
18-
modifyConfigurationStandardSequencesServicesAddProcessAccelerators_ = (gpu | pixelNtupletFit).makeProcessModifier(_addProcessAccelerators)
21+
from Configuration.ProcessModifiers.alpaka_cff import alpaka
22+
23+
modifyConfigurationStandardSequencesServicesAddProcessAccelerators_ = (gpu | pixelNtupletFit | alpaka).makeProcessModifier(_addProcessAccelerators)
24+
modifyConfigurationStandardSequencesServicesAddProcessAcceleratorsAlpaka_ = alpaka.makeProcessModifier(_addProcessAcceleratorsAlpaka)
1925

2026
# load TritonService when SONIC workflow is enabled
2127
def _addTritonService(process):

0 commit comments

Comments
 (0)