diff --git a/loopy/codegen/loop.py b/loopy/codegen/loop.py index 6d0b2ca60504060e677d1e3bb941b6c3ccbb3fc2..cc45f95e7548945d83f38ae0e7bc9c08c5d4f013 100644 --- a/loopy/codegen/loop.py +++ b/loopy/codegen/loop.py @@ -424,7 +424,8 @@ def generate_sequential_loop_dim_code(kernel, sched_index, codegen_state): from cgen import Comment result.append(Comment(cmt)) - from cgen import Initializer, POD, Const, Line + from loopy.codegen import POD + from cgen import Initializer, Const, Line from loopy.symbolic import aff_to_expr if (static_ubound - static_lbound).plain_is_zero():