From 35701a551515a1df905e4d5c433308077d823ace Mon Sep 17 00:00:00 2001
From: Andreas Kloeckner <inform@tiker.net>
Date: Sun, 21 Jul 2013 01:12:42 -0400
Subject: [PATCH] Guard against Nones in integer arg finder
---
loopy/compiled.py | 3 +++
1 file changed, 3 insertions(+)
diff --git a/loopy/compiled.py b/loopy/compiled.py
index e722fd338..bda857b22 100644
--- a/loopy/compiled.py
+++ b/loopy/compiled.py
@@ -146,6 +146,9 @@ def generate_integer_arg_finding_from_shapes(gen, kernel, impl_arg_info, flags):
if arg.arg_class is GlobalArg:
sym_shape = var(arg.name).attr("shape")
for axis_nr, shape_i in enumerate(arg.shape):
+ if shape_i is None:
+ continue
+
deps = dep_map(shape_i)
if len(deps) == 1:
--
GitLab