From b78f196de570e75f1cdad7f23d5ac4a482814e31 Mon Sep 17 00:00:00 2001 From: Tim Warburton Date: Tue, 1 Nov 2011 23:38:02 -0500 Subject: [PATCH] Expose loopy.generate_code. --- loopy/__init__.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/loopy/__init__.py b/loopy/__init__.py index 3e686fff0..a14dff965 100644 --- a/loopy/__init__.py +++ b/loopy/__init__.py @@ -26,11 +26,13 @@ from loopy.kernel import AutoFitLocalIndexTag from loopy.cse import realize_cse from loopy.preprocess import preprocess_kernel from loopy.schedule import generate_loop_schedules +from loopy.codegen import generate_code from loopy.compiled import CompiledKernel, drive_timing_run, auto_test_vs_seq from loopy.check import check_kernels __all__ = ["ScalarArg", "ArrayArg", "ImageArg", "preprocess_kernel", "generate_loop_schedules", + "generate_code", "CompiledKernel", "drive_timing_run", "check_kernels", "make_kernel", "split_dimension", "join_dimensions", "tag_dimensions", "realize_cse", "add_prefetch" -- GitLab