diff --git a/src/pystencils/types/types.py b/src/pystencils/types/types.py index ae0a8829df43135be99acec9abb3500f80a9abf5..2f0f2ff46ec1498eda724f23964bf28aa2ad2a9b 100644 --- a/src/pystencils/types/types.py +++ b/src/pystencils/types/types.py @@ -483,7 +483,20 @@ class PsIntegerType(PsScalarType, ABC): if not isinstance(value, np_dtype): raise PsTypeError(f"Given value {value} is not of required type {np_dtype}") unsigned_suffix = "" if self.signed else "u" - return f"(({self._c_type_without_const()}) {value}{unsigned_suffix})" + + match self.width: + case w if w < 32: + # Plain integer literals get at least type `int`, which is 32 bit in all relevant cases + # So we need to explicitly cast to smaller types + return f"(({self._c_type_without_const()}) {value}{unsigned_suffix})" + case 32: + # No suffix here - becomes `int`, which is 32 bit + return f"{value}{unsigned_suffix}" + case 64: + # LL suffix: `long long` is the only type guaranteed to be 64 bit wide + return f"{value}{unsigned_suffix}LL" + case _: + assert False, "unreachable code" def create_constant(self, value: Any) -> Any: np_type = self.NUMPY_TYPES[self._width] diff --git a/tests/nbackend/test_extensions.py b/tests/nbackend/test_extensions.py index 8d600ef766d55c51e93c23ae5bd720c03130b555..16e610a552b426cc4245e2e5c4ee36663f6c2bfa 100644 --- a/tests/nbackend/test_extensions.py +++ b/tests/nbackend/test_extensions.py @@ -54,6 +54,6 @@ def test_literals(): print(code) assert "const double x = C;" in code - assert "CELLS[((int64_t) 0)]" in code - assert "CELLS[((int64_t) 1)]" in code - assert "CELLS[((int64_t) 2)]" in code + assert "CELLS[0LL]" in code + assert "CELLS[1LL]" in code + assert "CELLS[2LL]" in code