From 8dc8808f93d6abc1eb5a31dff16387d75f6d8f78 Mon Sep 17 00:00:00 2001
From: Andreas Kloeckner <inform@tiker.net>
Date: Wed, 12 Feb 2025 09:47:41 -0600
Subject: [PATCH] get_access_info typing fix: vec_info may be None
---
loopy/kernel/array.py | 9 +++++----
1 file changed, 5 insertions(+), 4 deletions(-)
diff --git a/loopy/kernel/array.py b/loopy/kernel/array.py
index 9895685f..5844143f 100644
--- a/loopy/kernel/array.py
+++ b/loopy/kernel/array.py
@@ -1217,10 +1217,11 @@ def _apply_offset(sub: Expression, ary: ArrayBase) -> Expression:
def get_access_info(kernel: LoopKernel,
- ary: ArrayArg | TemporaryVariable,
- index: Expression | tuple[Expression, ...],
- eval_expr: Callable[[Expression], int],
- vectorization_info: VectorizationInfo) -> AccessInfo:
+ ary: ArrayArg | TemporaryVariable,
+ index: Expression | tuple[Expression, ...],
+ eval_expr: Callable[[Expression], int],
+ vectorization_info: VectorizationInfo | None
+ ) -> AccessInfo:
"""
:arg ary: an object of type :class:`ArrayBase`
:arg index: a tuple of indices representing a subscript into ary
--
GitLab