diff --git a/doc/reference.rst b/doc/reference.rst index c8d04107931576465df7046dde11a0bed7cbe8b3..bf62ec89ed9626b8fc1976b266d03784e3c9be1f 100644 --- a/doc/reference.rst +++ b/doc/reference.rst @@ -242,6 +242,13 @@ Finishing up .. autofunction:: generate_code +Running +------- + +.. autoclass:: InvocationFlags + +.. autoclass:: CompiledKernel + Automatic Testing ----------------- diff --git a/loopy/__init__.py b/loopy/__init__.py index a2d1f971d8ea71b11dbdf2e849843a1e5d381b65..e1a5a6da34d2f5b5ab23e4797ede770a5c3b27b7 100644 --- a/loopy/__init__.py +++ b/loopy/__init__.py @@ -68,7 +68,7 @@ from loopy.preprocess import (preprocess_kernel, realize_reduction, infer_unknown_types) from loopy.schedule import generate_loop_schedules from loopy.codegen import generate_code -from loopy.compiled import CompiledKernel +from loopy.compiled import InvocationFlags, CompiledKernel from loopy.auto_test import auto_test_vs_ref __all__ = [ @@ -88,7 +88,10 @@ __all__ = [ "preprocess_kernel", "realize_reduction", "infer_unknown_types", "generate_loop_schedules", "generate_code", - "CompiledKernel", "auto_test_vs_ref", + + "InvocationFlags", "CompiledKernel", + + "auto_test_vs_ref", "make_kernel", "split_iname", "join_inames", "tag_inames", "duplicate_inames",