From cd438f9523c87da8e35f2e7d135283df2a10be3c Mon Sep 17 00:00:00 2001
From: Stephan Seitz <stephan.seitz@fau.de>
Date: Mon, 16 Dec 2019 14:24:34 +0100
Subject: [PATCH] Don't use CUDA for doctest
---
docs/index.rst | 4 ++--
1 file changed, 2 insertions(+), 2 deletions(-)
diff --git a/docs/index.rst b/docs/index.rst
index 946234c..b940ae7 100644
--- a/docs/index.rst
+++ b/docs/index.rst
@@ -84,7 +84,7 @@ You can also use the class :class:`pystencils_autodiff.AutoDiffOp` to obtain bot
op = AutoDiffOp(forward_assignments)
backward_assignments = op.backward_assignments
- tensorflow_op = op.create_tensorflow_op(backend='tensorflow_native')
+ tensorflow_op = op.create_tensorflow_op(backend='tensorflow_native', use_cuda=False)
.. testoutput::
:hide:
@@ -97,7 +97,7 @@ You can also use the class :class:`pystencils_autodiff.AutoDiffOp` to obtain bot
.. testcode::
- torch_op = op.create_tensorflow_op(backend='torch_native')
+ torch_op = op.create_tensorflow_op(backend='torch_native', use_cuda=False)
Contents
--
GitLab