From 2d6f6c4da10f47a6705be13cc7e6af5065e588ec Mon Sep 17 00:00:00 2001
From: Andreas Kloeckner <inform@tiker.net>
Date: Sun, 28 Feb 2016 21:52:12 -0600
Subject: [PATCH] generate_code: call prior codegen steps if not done already
---
loopy/codegen/__init__.py | 7 ++++++-
1 file changed, 6 insertions(+), 1 deletion(-)
diff --git a/loopy/codegen/__init__.py b/loopy/codegen/__init__.py
index 2e136d7bb..91af52cde 100644
--- a/loopy/codegen/__init__.py
+++ b/loopy/codegen/__init__.py
@@ -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")
--
GitLab