From 9e4b6b7476456574f9595e57e4976f39572c01b7 Mon Sep 17 00:00:00 2001 From: Andreas Kloeckner <inform@tiker.net> Date: Wed, 18 Feb 2015 15:13:07 -0600 Subject: [PATCH] Re-Fix invoker code generation for shape=None --- loopy/compiled.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/loopy/compiled.py b/loopy/compiled.py index 02730d868..c5928d6e7 100644 --- a/loopy/compiled.py +++ b/loopy/compiled.py @@ -502,7 +502,7 @@ def generate_array_arg_setup(gen, kernel, impl_arg_info, options): if kernel_arg.shape is None: pass - if any(shape_axis is None for shape_axis in kernel_arg.shape): + elif any(shape_axis is None for shape_axis in kernel_arg.shape): gen("if len(%s.shape) != %s:" % (arg.name, len(arg.unvec_shape))) with Indentation(gen): -- GitLab