Improve nLevels checking for Integer

Author: Xeratec

Deeploy/AbstractDataTypes.py

@@ -209,6 +209,10 @@ def typeMin(cls) -> int:
         else:
             return 0
 
+    @_classproperty
+    def nLevels(cls) -> int:
+        return cls.typeMax - cls.typeMin + 1
+
     @classmethod
     def partialOrderUpcast(cls, otherCls: Type[Immediate]) -> bool:
         if issubclass(otherCls, IntegerImmediate):
@@ -234,6 +238,10 @@ def checkValue(cls, value: Union[int, Iterable[int], np.ndarray], ctxt: Optional
             return False
         return True
 
+    @classmethod
+    def checkNumLevels(cls, nLevels: int, signed: bool) -> bool:
+        return nLevels <= (cls.typeMax - cls.typeMin + 1)
+
 
 class FloatImmediate(Immediate[Union[float, Iterable[float]], _ImmediateType]):
     typeMantissa: int  #: int: Represents the number of bits reserved for the mantissa part
@@ -353,6 +361,10 @@ def checkPromotion(cls, _value: Union[Optional[str], Pointer], ctxt: Optional[_N
             value = _value
         return cls.checkValue(value, ctxt)
 
+    @classmethod
+    def checkNumLevels(cls, nLevels: int, signed: bool) -> bool:
+        return cls.referencedType.checkNumLevels(nLevels, signed)
+
     def __init__(self, _value: Union[Optional[str], Pointer], ctxt: Optional[_NetworkContext] = None):
         """Initializes a pointer to a registered object in the NetworkContext
 

Deeploy/CommonExtensions/TypeCheckers/SignPropTypeChecker.py

@@ -27,6 +27,7 @@
 
 import onnx_graphsurgeon as gs
 
+from Deeploy.AbstractDataTypes import IntegerImmediate
 from Deeploy.DeeployTypes import ConstantBuffer, NetworkContext, NodeTypeChecker, OperatorRepresentation, VariableBuffer
 
 
@@ -66,9 +67,14 @@ def typeInferOutput(self, ctxt: NetworkContext, node: gs.Node,
         if signProp:
             nLevels = self._inferNumLevels(inputs, operatorRepresentation)
             signedness = self._inferSignedness(inputs, operatorRepresentation)
-
-            for obj, nLevels, sign in zip(outputs, nLevels, signedness):
-                obj.nLevels = nLevels
+            for obj, nLevel, sign in zip(outputs, nLevels, signedness):
+                obj.nLevels = nLevel
                 obj._signed = sign
 
+                if issubclass(obj._type.referencedType,
+                              IntegerImmediate) and not obj._type.checkNumLevels(nLevel, sign):
+                    print(
+                        f"[WARNING] {obj.name} has {nLevel} levels, but {obj._type.referencedType.typeName} only supports {obj._type.referencedType.nLevels} levels."
+                    )
+
         return ctxt