From 703b31d128e9aed582fb0c3f2402f82745e48352 Mon Sep 17 00:00:00 2001
From: Andreas Kloeckner <inform@tiker.net>
Date: Thu, 25 May 2017 09:34:20 -0700
Subject: [PATCH] Fix array offset treatment if offset is None
---
loopy/kernel/array.py | 5 ++---
1 file changed, 2 insertions(+), 3 deletions(-)
diff --git a/loopy/kernel/array.py b/loopy/kernel/array.py
index 40612dc83..d9f2faf7c 100644
--- a/loopy/kernel/array.py
+++ b/loopy/kernel/array.py
@@ -1208,10 +1208,9 @@ def get_access_info(target, ary, index, eval_expr, vectorization_info):
if ary.offset:
if ary.offset is lp.auto:
- offset_name = array_name+"_offset"
+ return var(array_name+"_offset") + sub
elif isinstance(ary.offset, str):
- offset_name = ary.offset
- return var(offset_name) + sub
+ return var(ary.offset) + sub
else:
# assume it's an expression
return ary.offset + sub
--
GitLab