cudaPreallocate Namespace Reference

Functions
def	main

Variables
tuple	opts = parser.parse_args()
tuple	parser

Function Documentation

def cudaPreallocate.main

(

opts

)

Definition at line 8 of file cudaPreallocate.py.

References join(), and print().

def main(opts):
    device = []
    host = []

    device_re = re.compile("Device.*allocated new device block.*\((?P<bytes>\d+) bytes")
    host_re = re.compile("Host.*allocated new host block.*\((?P<bytes>\d+) bytes")

    f = open(opts.file)
    for line in f:
        m = device_re.search(line)
        if m:
            device.append(m.group("bytes"))
            continue
        m = host_re.search(line)
        if m:
            host.append(m.group("bytes"))
    f.close()

    print("process.CUDAService.allocator.devicePreallocate = cms.untracked.vuint32(%s)" % ",".join(device))
    print("process.CUDAService.allocator.hostPreallocate = cms.untracked.vuint32(%s)" % ",".join(host))