From 11f45c4988ec55dd0128c5b89e570cf60c1b47a0 Mon Sep 17 00:00:00 2001
From: Stephan Seitz <stephan.seitz@fau.de>
Date: Fri, 28 Feb 2020 13:30:53 +0100
Subject: [PATCH] Fix test_module_printing_globals
---
tests/test_module_printing.py | 7 +------
1 file changed, 1 insertion(+), 6 deletions(-)
diff --git a/tests/test_module_printing.py b/tests/test_module_printing.py
index d94bbef..ad3c20a 100644
--- a/tests/test_module_printing.py
+++ b/tests/test_module_printing.py
@@ -92,13 +92,9 @@ def test_module_printing_globals():
z[0, 0]: x[0, 0] * sympy.log(TextureCachedField(x).at(sympy.Matrix((0.43, 3))) * y[0, 0])
})
- backward_assignments = create_backward_assignments(forward_assignments)
-
forward_ast = pystencils.create_kernel(forward_assignments, target)
forward_ast.function_name = 'forward'
- backward_ast = pystencils.create_kernel(backward_assignments, target)
- backward_ast.function_name = 'backward'
- module = TorchModule("hallo", [forward_ast, backward_ast])
+ module = TorchModule("hallo", [forward_ast])
print(module)
@@ -122,4 +118,3 @@ def test_custom_printer():
backward_ast.function_name = 'backward'
module = TorchModule("hallo", [forward_ast, backward_ast])
print(DoesNotLikeTorchPrinter()(module))
-
--
GitLab