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

Fix test_module_printing_globals

parent 0e9f33bd
No related branches found
No related tags found
No related merge requests found
......@@ -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))
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment