diff --git a/loopy/__init__.py b/loopy/__init__.py index 3e686fff006baafc17c3195c8f7127bc4e30d801..a14dff965bd349bac80bbda3e6deb6bb67e2ae5a 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"