Skip to content
Snippets Groups Projects
Commit 2d6f6c4d authored by Andreas Klöckner's avatar Andreas Klöckner
Browse files

generate_code: call prior codegen steps if not done already

parent 6e176aed
No related branches found
No related tags found
No related merge requests found
......@@ -464,10 +464,15 @@ def generate_code(kernel, device=None):
warn("passing 'device' to generate_code() is deprecated",
DeprecationWarning, stacklevel=2)
from loopy.kernel import kernel_state
if kernel.state == kernel_state.INITIAL:
from loopy.preprocess import preprocess_kernel
kernel = preprocess_kernel(kernel)
if kernel.schedule is None:
from loopy.schedule import get_one_scheduled_kernel
kernel = get_one_scheduled_kernel(kernel)
from loopy.kernel import kernel_state
if kernel.state != kernel_state.SCHEDULED:
raise LoopyError("cannot generate code for a kernel that has not been "
"scheduled")
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment