diff --git a/pykokkos/core/fusion/access_modes.py b/pykokkos/core/fusion/access_modes.py index bea5321..1b392e7 100644 --- a/pykokkos/core/fusion/access_modes.py +++ b/pykokkos/core/fusion/access_modes.py @@ -132,6 +132,10 @@ def visit_Subscript(self, node: ast.Subscript) -> None: slices.insert(0, index) current_node = current_node.value + # The subscript itself could be indexing another view + for s in slices: + self.visit(s) + # Avoid type annotations if isinstance(current_node, ast.Attribute): return