HeterogeneousTest/CUDAKernel/test/testCUDATestKernelAdditionModule.py - master - githubmirror/cmssw

HeterogeneousTest/CUDAKernel/test/testCUDATestKernelAdditionModule.py

14 строк430 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('TestCUDATestKernelAdditionModule')
process.load('HeterogeneousCore.CUDACore.ProcessAcceleratorCUDA_cfi')

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

process.cudaTestKernelAdditionModule = cms.EDAnalyzer('CUDATestKernelAdditionModule',
    size = cms.uint32( 1024*1024 )
)

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

process.maxEvents.input = 1