diff --git a/examples/python/hello-loopy.py b/examples/python/hello-loopy.py index 82ff2e60dee345fe16771d09cb39d2d56e9f493d..7c5de5a1b1d7042498a12204959a59021ac5e0d8 100644 --- a/examples/python/hello-loopy.py +++ b/examples/python/hello-loopy.py @@ -26,5 +26,5 @@ knl = lp.split_iname(knl, "i", 128, outer_tag="g.0", inner_tag="l.0") evt, (out,) = knl(queue, a=a) # ENDEXAMPLE -cknl = lp.CompiledKernel(ctx, knl) -print(cknl.get_highlighted_code({"a": np.float32})) +knl = lp.add_and_infer_dtypes(knl, {"a": np.dtype(np.float32)}) +print(lp.generate_code_v2(knl).device_code())