HeterogeneousTest/AlpakaWrapper/test/testAlpakaTestWrapperAdditionModule.py - master - githubmirror/cmssw

HeterogeneousTest/AlpakaWrapper/test/testAlpakaTestWrapperAdditionModule.py

15 строк518 B

Andrea Bocci

Add a test for Alpaka libraries and build rules

05 апр 2024, 23:44

05 апр 2024, 23:44f963c4c

import FWCore.ParameterSet.Config as cms

process = cms.Process('TestAlpakaTestWrapperAdditionModule')
process.load('Configuration.StandardSequences.Accelerators_cff')
process.load('HeterogeneousCore.AlpakaCore.ProcessAcceleratorAlpaka_cfi')

process.source = cms.Source('EmptySource')

process.alpakaTestWrapperAdditionModule = cms.EDAnalyzer('AlpakaTestWrapperAdditionModule@alpaka',
    size = cms.uint32( 1024*1024 )
)

process.path = cms.Path(process.alpakaTestWrapperAdditionModule)

process.maxEvents.input = 1