diff --git a/loopy/kernel/array.py b/loopy/kernel/array.py index 40612dc838cb2e6299618ba58ad5f816fc3d146c..d9f2faf7cff43ca6aac91520c0e096be209d5c21 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