From cc63afdc7aa664d61be1f60727d1d899edd6ae60 Mon Sep 17 00:00:00 2001 From: Matthias Diener Date: Wed, 30 Mar 2022 18:42:23 -0500 Subject: [PATCH] demo.py: add loopy code, use #! --- examples/demo.py | 10 +++++++++- 1 file changed, 9 insertions(+), 1 deletion(-) diff --git a/examples/demo.py b/examples/demo.py index 32017cb..ce6f2c5 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 ===============") # }}} -- GitLab