Skip to content
Snippets Groups Projects
Commit cd438f95 authored by Stephan Seitz's avatar Stephan Seitz
Browse files

Don't use CUDA for doctest

parent 75f1a1c0
No related branches found
No related tags found
No related merge requests found
Pipeline #20549 passed
......@@ -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
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment