HeterogeneousTest/CUDAOpaque/test/testCUDATestAdditionModules.py - master - githubmirror/cmssw

HeterogeneousTest/CUDAOpaque/test/testCUDATestAdditionModules.py

31 строка1 KB

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('TestCUDATestOpaqueAdditionModule')
process.load('HeterogeneousCore.CUDACore.ProcessAcceleratorCUDA_cfi')

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

process.cudaTestDeviceAdditionModule = cms.EDAnalyzer('CUDATestDeviceAdditionModule',
    size = cms.uint32( 1024*1024 )
)

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

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

process.cudaTestOpaqueAdditionModule = cms.EDAnalyzer('CUDATestOpaqueAdditionModule',
    size = cms.uint32( 1024*1024 )
)

process.path = cms.Path(
    # this one fails with "cudaErrorInvalidDeviceFunction: invalid device function"
    #process.cudaTestDeviceAdditionModule +
    process.cudaTestKernelAdditionModule +
    process.cudaTestWrapperAdditionModule +
    process.cudaTestOpaqueAdditionModule)

process.maxEvents.input = 1