Merge pull request #4217 from kmaehashi/fix-multi-gpu-doctest

raise friendly error for multi-gpu doctest failures
chainer · Jan 24, 2018 · bc36bf2 · bc36bf2
2 parents 0080f6a + 16cacf9
commit bc36bf2
Showing 1 changed file with 8 additions and 0 deletions.
diff --git a/docs/source/tutorial/gpu.rst b/docs/source/tutorial/gpu.rst
@@ -17,6 +17,14 @@ After reading this section, you will be able to:
 * Write model-parallel computing in Chainer
 * Write data-parallel computing in Chainer
 
+.. testcode::
+   :hide:
+
+   try:
+       with cupy.cuda.Device(1):
+           pass
+   except cupy.cuda.runtime.CUDARuntimeError:
+       raise RuntimeError('doctest in this document requires 2 GPUs') from None
 
 Relationship between Chainer and CuPy
 ~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~