HeterogeneousTest/CUDAWrapper/test/testCUDATestWrapperAdditionModule.py - master - githubmirror/cmssw

HeterogeneousTest/CUDAWrapper/test/testCUDATestWrapperAdditionModule.py

14 строк434 B

Andrea Bocci

Add abstract interfaces for the CUDA and ROCm services

28 фев 2023, 15:35

28 фев 2023, 15:355ebe8d8

import FWCore.ParameterSet.Config as cms

process = cms.Process('TestCUDATestWrapperAdditionModule')
process.load('HeterogeneousCore.CUDACore.ProcessAcceleratorCUDA_cfi')

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

process.cudaTestWrapperAdditionModule = cms.EDAnalyzer('CUDATestWrapperAdditionModule',
    size = cms.uint32( 1024*1024 )
)

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

process.maxEvents.input = 1