From 35764348ad9fd6d5f778d69692b0212fd9111630 Mon Sep 17 00:00:00 2001 From: Frederik Hennig <frederik.hennig@fau.de> Date: Fri, 7 Feb 2025 15:24:22 +0100 Subject: [PATCH] fix cstdint collection --- src/pystencilssfg/lang/expressions.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/src/pystencilssfg/lang/expressions.py b/src/pystencilssfg/lang/expressions.py index b3ed18e..72287ea 100644 --- a/src/pystencilssfg/lang/expressions.py +++ b/src/pystencilssfg/lang/expressions.py @@ -7,7 +7,7 @@ import sympy as sp from pystencils import TypedSymbol from pystencils.codegen import Parameter -from pystencils.types import PsType, PsScalarType, UserTypeSpec, create_type +from pystencils.types import PsType, PsIntegerType, UserTypeSpec, create_type from ..exceptions import SfgException from .headers import HeaderFile @@ -474,7 +474,7 @@ def includes(obj: ExprLike | PsType) -> set[HeaderFile]: case PsType(): headers = set(HeaderFile.parse(h) for h in obj.required_headers) - if isinstance(obj, PsScalarType): + if isinstance(obj, PsIntegerType): headers.add(HeaderFile.parse("<cstdint>")) return headers -- GitLab