AlpakaServices/test/testAlpakaServiceCudaAsync.py

0001 import FWCore.ParameterSet.Config as cms
0002
0003 process = cms.Process( "TEST" )
0004
0005 process.options = cms.untracked.PSet(
0006     numberOfThreads = cms.untracked.uint32( 4 ),
0007     numberOfStreams = cms.untracked.uint32( 0 ),
0008 )
0009
0010 process.load('FWCore.MessageService.MessageLogger_cfi')
0011 process.MessageLogger.CUDAService = {}
0012 process.MessageLogger.AlpakaService = {}
0013
0014 process.load('HeterogeneousCore.CUDAServices.CUDAService_cfi')
0015
0016 from HeterogeneousCore.AlpakaServices.AlpakaServiceCudaAsync_cfi import AlpakaServiceCudaAsync as _AlpakaServiceCudaAsync
0017 process.AlpakaServiceCudaAsync = _AlpakaServiceCudaAsync.clone(
0018     verbose = True,
0019     hostAllocator = dict(
0020       binGrowth = 2,
0021       minBin = 8,                           # 256 bytes
0022       maxBin = 30,                          #   1 GB
0023       maxCachedBytes = 64*1024*1024*1024,   #  64 GB
0024       maxCachedFraction = 0.8,              # or 80%, whatever is less
0025       fillAllocations = True,
0026       fillAllocationValue = 0xA5,
0027       fillReallocations = True,
0028       fillReallocationValue = 0x69,
0029       fillDeallocations = True,
0030       fillDeallocationValue = 0x5A,
0031       fillCaches = True,
0032       fillCacheValue = 0x96
0033     ),
0034     deviceAllocator = dict(
0035       binGrowth = 2,
0036       minBin = 8,                           # 256 bytes
0037       maxBin = 30,                          #   1 GB
0038       maxCachedBytes = 8*1024*1024*1024,    #   8 GB
0039       maxCachedFraction = 0.8,              # or 80%, whatever is less
0040       fillAllocations = True,
0041       fillAllocationValue = 0xA5,
0042       fillReallocations = True,
0043       fillReallocationValue = 0x69,
0044       fillDeallocations = True,
0045       fillDeallocationValue = 0x5A,
0046       fillCaches = True,
0047       fillCacheValue = 0x96
0048     )
0049 )
0050
0051 process.source = cms.Source("EmptySource")
0052
0053 process.maxEvents = cms.untracked.PSet(
0054     input = cms.untracked.int32( 0 )
0055 )