Use multiple GPUs in test
@@ -38,17 +38,18 @@ class TestMonteCarlo(unittest.TestCase):
class TestMonteCarloWithMultiGPU(unittest.TestCase):
- def test_monte_carlo(self):
+ def test_monte_carlo_multigpu(self):
output = example_test.run_example(
- 'finance/monte_carlo_multigpu.py', '--n-options', '10',
+ 'finance/monte_carlo_multigpu.py', '--gpus', '0', '1',
+ '--n-options', '10',
'--n-samples-per-thread', '10',
'--n-threads-per-option', '10')
six.assertRegex(
self, output.decode('utf-8'),
r'initializing...\n' +
r'start computation\n' +
- r' # of gpus: 1\n' +
+ r' # of gpus: 2\n' +
r' # of options: 10\n' +
- r' # of samples per option: 100\n' +
+ r' # of samples per option: 200\n' +
r'GPU \(CuPy, Monte Carlo method\):\t[0-9\.]+ sec\n' +
r'Error: [0-9\.]+')
