From 8532ccdbda7e70a80cfc54e3658d97f985848030 Mon Sep 17 00:00:00 2001
From: Frederik Hennig <frederik.hennig@fau.de>
Date: Tue, 22 Oct 2024 14:20:43 +0200
Subject: [PATCH] fix: PsBufferAcc is a legal argument to PsAddressOf
---
src/pystencils/backend/kernelcreation/typification.py | 4 ++--
1 file changed, 2 insertions(+), 2 deletions(-)
diff --git a/src/pystencils/backend/kernelcreation/typification.py b/src/pystencils/backend/kernelcreation/typification.py
index debcc3cf6..04aeddabe 100644
--- a/src/pystencils/backend/kernelcreation/typification.py
+++ b/src/pystencils/backend/kernelcreation/typification.py
@@ -472,7 +472,7 @@ class Typifier:
self._handle_idx(idx)
case PsAddressOf(arg):
- if not isinstance(arg, (PsSymbolExpr, PsSubscript, PsMemAcc, PsLookup)):
+ if not isinstance(arg, (PsSymbolExpr, PsSubscript, PsMemAcc, PsBufferAcc, PsLookup)):
raise TypificationError(
f"Illegal expression below AddressOf operator: {arg}"
)
@@ -489,7 +489,7 @@ class Typifier:
match arg:
case PsSymbolExpr(s):
pointed_to_type = s.get_dtype()
- case PsSubscript(ptr, _) | PsMemAcc(ptr, _):
+ case PsSubscript(ptr, _) | PsMemAcc(ptr, _) | PsBufferAcc(ptr, _):
arr_type = ptr.get_dtype()
assert isinstance(arr_type, PsDereferencableType)
pointed_to_type = arr_type.base_type
--
GitLab