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

Add generate_shared_object test for torch

parent ecf5c1b7
Branches
No related merge requests found
Pipeline #22328 failed with stage
in 12 seconds
......@@ -31,3 +31,13 @@ def test_wrap_tensorflow():
generate_shared_object(tempfile.TemporaryDirectory, None, show_code=True,
framework_module_class=TensorflowModule, generate_code_only=True)
def test_wrap_torch():
import pytest
pytest.importorskip("pystencils_autodiff")
from pystencils_autodiff.backends.astnodes import TorchModule
generate_shared_object(tempfile.TemporaryDirectory, None, show_code=True,
framework_module_class=TorchModule, generate_code_only=True)
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