chainer.testing.attr.multi_gpu

chainer.testing.attr.multi_gpu(gpu_num)[source]

Decorator to indicate number of GPUs required to run the test.

Tests can be annotated with this decorator (e.g., @multi_gpu(2)) to declare number of GPUs required to run. When running tests, if CHAINER_TEST_GPU_LIMIT environment variable is set to value greater than or equals to 0, test cases that require GPUs more than the limit will be skipped.