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 merge requests found
Pipeline #20549 passed with stage
in 5 minutes and 26 seconds
......@@ -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% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment