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

Fix test custom printer

parent 96b29fb1
No related branches found
No related tags found
No related merge requests found
Pipeline #22353 failed
Stage: test
Hide whitespace changes
Inline Side-by-side
tests/test_module_printing.py
+ 1
− 5
View file @ e044a9cf
...@@ -110,11 +110,7 @@ def test_custom_printer(): ...@@ -110,11 +110,7 @@ def test_custom_printer():
z[0, 0]: x[0, 0] * sympy.log(TextureCachedField(x).at(sympy.Matrix((0.43, 3))) * y[0, 0]) 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) forward_ast = pystencils.create_kernel(forward_assignments)
forward_ast.function_name = 'forward' forward_ast.function_name = 'forward'
backward_ast = pystencils.create_kernel(backward_assignments) module = TorchModule("hallo", [forward_ast])
backward_ast.function_name = 'backward'
module = TorchModule("hallo", [forward_ast, backward_ast])
print(DoesNotLikeTorchPrinter()(module)) print(DoesNotLikeTorchPrinter()(module))
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment