HeterogeneousTest/ROCmKernel/test/testROCmTestKernelAdditionModule.py - master - githubmirror/cmssw

HeterogeneousTest/ROCmKernel/test/testROCmTestKernelAdditionModule.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('TestROCmTestKernelAdditionModule')
process.load('HeterogeneousCore.ROCmCore.ProcessAcceleratorROCm_cfi')

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

process.rocmTestKernelAdditionModule = cms.EDAnalyzer('ROCmTestKernelAdditionModule',
    size = cms.uint32( 1024*1024 )
)

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

process.maxEvents.input = 1