diff --git a/src/pystencils/backend/kernelcreation/typification.py b/src/pystencils/backend/kernelcreation/typification.py index ec615d8f4bdf7276fe6699ca0837cfc9b4fc2487..190cd9e23b715ac5bca2a4bc6fd119013376cb49 100644 --- a/src/pystencils/backend/kernelcreation/typification.py +++ b/src/pystencils/backend/kernelcreation/typification.py @@ -184,6 +184,11 @@ class TypeContext: expr.dtype = self._target_type def _compatible(self, dtype: PsType): + """Checks whether the given data type is compatible with the context's target type. + + If the target type is ``const``, they must be equal up to const qualification; + if the target type is not ``const``, `dtype` must match it exactly. + """ assert self._target_type is not None if self._target_type.const: return constify(dtype) == self._target_type