Add FloatAddChecker to CI (and make it work again, if possible)

pulp-platform · Nov 13, 2024 · 35d2fa5 · 35d2fa5
1 parent b6cf6fa
commit 35d2fa5
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 2 deletions.
diff --git a/.github/workflows/CI.yml b/.github/workflows/CI.yml
@@ -48,6 +48,7 @@ jobs:
         testRequantizedDWConv
         test2DRequantizedConv
         iSoftmax
+        testFloatAdder
 
   generic-models:
     uses: ./.github/workflows/TestRunnerGeneric.yml
@@ -149,6 +150,7 @@ jobs:
         Hardswish
         RQHardswish
         testBacktracking
+        testFloatAdder
       num-cores: 8
 
   siracusa-models:

diff --git a/Deeploy/Targets/Generic/TypeCheckers.py b/Deeploy/Targets/Generic/TypeCheckers.py
@@ -31,7 +31,7 @@
 
 from Deeploy.AbstractDataTypes import Pointer
 from Deeploy.CommonExtensions.TypeCheckers.SignPropTypeChecker import SignPropTypeChecker
-from Deeploy.DeeployTypes import ConstantBuffer, NodeTypeChecker, OperatorRepresentation, VariableBuffer
+from Deeploy.DeeployTypes import ConstantBuffer, OperatorRepresentation, VariableBuffer
 
 
 class ConcatChecker(SignPropTypeChecker):
@@ -125,11 +125,19 @@ def _inferSignedness(self, inputs: List[VariableBuffer],
             return [False]
 
 
-class FloatAddChecker(NodeTypeChecker):
+class FloatAddChecker(SignPropTypeChecker):
 
     def __init__(self, input_types: Sequence[Type[Pointer]], output_types: Sequence[Type[Pointer]]):
         super().__init__(input_types, output_types)
 
+    def _inferNumLevels(self, inputs: List[VariableBuffer],
+                        operatorRepresentation: OperatorRepresentation) -> List[int]:
+        return [inputs[0].nLevels + inputs[1].nLevels]
+
+    def _inferSignedness(self, inputs: List[VariableBuffer],
+                         operatorRepresentation: OperatorRepresentation) -> List[bool]:
+        return [True]
+
 
 class GatherChecker(SignPropTypeChecker):