diff --git a/examples/demo.py b/examples/demo.py index 32017cbb455c62afe2aa923dc173007ee8bf1fd9..ce6f2c5ac37f4035a0a700bd7cf82a133365e136 100644 --- a/examples/demo.py +++ b/examples/demo.py @@ -1,3 +1,5 @@ +#!/usr/bin/env python + import numpy as np import pytato as pt @@ -22,11 +24,17 @@ assert np.allclose(out["aat"], a@a.T) # }}} -# {{{ generate OpenCL code +# {{{ print generated Loopy and OpenCL code prg = pt.generate_loopy(result) +print("=============== Loopy code =================") +print(prg.program) +print("============= End Loopy code ===============") + import loopy as lp +print("============== OpenCL code =================") print(lp.generate_code_v2(prg.program).device_code()) +print("============ End OpenCL code ===============") # }}}