From 0376f6f80b63da83aca526ba1a3d3f185e0c0d14 Mon Sep 17 00:00:00 2001
From: Stephan Seitz <stephan.seitz@fau.de>
Date: Thu, 27 Feb 2020 18:20:43 +0100
Subject: [PATCH] Add generate_shared_object test for torch
---
tests/test_alternative_wrappers.py | 10 ++++++++++
1 file changed, 10 insertions(+)
diff --git a/tests/test_alternative_wrappers.py b/tests/test_alternative_wrappers.py
index afbf7af..3e215f8 100644
--- a/tests/test_alternative_wrappers.py
+++ b/tests/test_alternative_wrappers.py
@@ -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)
--
GitLab