diff options
Diffstat (limited to 'threadtest.py')
| -rw-r--r-- | threadtest.py | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/threadtest.py b/threadtest.py index 5dffb98..92b0c81 100644 --- a/threadtest.py +++ b/threadtest.py @@ -61,7 +61,7 @@ if __name__ == '__main__': parser = optparse.OptionParser('%prog') parser.add_option('-b', type='int', dest='nbits', default=8) - parser.add_option('-j', type='int', dest='ndevices', default=1) + parser.add_option('-j', type='string', dest='devices', default=1) parser.add_option('-n', type='int', dest='nblocks', default=64) options, args = parser.parse_args() @@ -74,7 +74,7 @@ if __name__ == '__main__': cuda.init() gputhreads = [] - for i in range(options.ndevices): + for i in [int(s) for s in options.devices.split(',')]: gputhreads.append(GPUThread(i, detector, jobs, output, options.nblocks)) gputhreads[-1].start() |