diff options
Diffstat (limited to 'threadtest.py')
| -rw-r--r-- | threadtest.py | 3 |
1 files changed, 2 insertions, 1 deletions
diff --git a/threadtest.py b/threadtest.py index 11bdaa4..fe90b4e 100644 --- a/threadtest.py +++ b/threadtest.py @@ -65,6 +65,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('-n', type='int', dest='nblocks', default=64) options, args = parser.parse_args() lbne = LBNE() @@ -74,7 +75,7 @@ if __name__ == '__main__': gputhreads = [] for i in range(options.ndevices): - gputhreads.append(GPUThread(i, lbne, jobs, output)) + gputhreads.append(GPUThread(i, lbne, jobs, output, options.nblocks)) gputhreads[-1].start() try: |