Commit 1d2c6653 authored by Stephan Seitz's avatar Stephan Seitz
Browse files

Call torch operation

parent cae9de91
Pipeline #22358 failed with stage
in 60 minutes and 53 seconds
...@@ -57,6 +57,7 @@ def test_torch_simple(): ...@@ -57,6 +57,7 @@ def test_torch_simple():
kernel = move(x, y, a).compile() kernel = move(x, y, a).compile()
pystencils.autodiff.show_code(kernel.ast) pystencils.autodiff.show_code(kernel.ast)
kernel().forward(x, y, 3)
def test_downsample(): def test_downsample():
......
Markdown is supported
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