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
Branches
Tags
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 ...@@ -84,7 +84,7 @@ You can also use the class :class:`pystencils_autodiff.AutoDiffOp` to obtain bot
op = AutoDiffOp(forward_assignments) op = AutoDiffOp(forward_assignments)
backward_assignments = op.backward_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:: .. testoutput::
:hide: :hide:
...@@ -97,7 +97,7 @@ You can also use the class :class:`pystencils_autodiff.AutoDiffOp` to obtain bot ...@@ -97,7 +97,7 @@ You can also use the class :class:`pystencils_autodiff.AutoDiffOp` to obtain bot
.. testcode:: .. testcode::
torch_op = op.create_tensorflow_op(backend='torch_native') torch_op = op.create_tensorflow_op(backend='torch_native', use_cuda=False)
Contents Contents
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment