Skip to content

Commit c0753ea

Browse files
committed
Removed unnecessary new ConvTranspose type checker. Added the generic Autoencoder1D test to the CI Pipeline
1 parent 3ab0478 commit c0753ea

File tree

3 files changed

+7
-26
lines changed

3 files changed

+7
-26
lines changed

.github/workflows/ci-platform-generic.yml

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -96,3 +96,4 @@ jobs:
9696
CCT/CCT_1_16_16_8
9797
CCT/CCT_2_32_32_128_Opset20
9898
testFloatDemoTinyViT
99+
Autoencoder1D

Deeploy/Targets/Generic/Bindings.py

Lines changed: 6 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -21,10 +21,10 @@
2121
RQIntegerDivTemplate, RQSiGELUTemplate, SliceTemplate, TransposeTemplate, iGELUTemplate, iLayernormTemplate, \
2222
iRMSNormTemplate, iSoftmaxTemplate
2323
from Deeploy.Targets.Generic.TypeCheckers import AddChecker, BatchNormChecker, ConcatChecker, ConvChecker, \
24-
ConvTransposeChecker, DebugPrintChecker, DequantChecker, DivChecker, DummyChecker, GatherChecker, GELUChecker, \
25-
GEMMChecker, LayerNormChecker, MatMulChecker, MaxPoolChecker, MulChecker, PadChecker, QuantChecker, \
26-
ReduceMeanChecker, ReduceSumChecker, ReluChecker, RequantShiftChecker, ReshapeChecker, RQIntegerDivChecker, \
27-
SliceChecker, SoftmaxChecker, TransposeChecker
24+
DebugPrintChecker, DequantChecker, DivChecker, DummyChecker, GatherChecker, GELUChecker, GEMMChecker, \
25+
LayerNormChecker, MatMulChecker, MaxPoolChecker, MulChecker, PadChecker, QuantChecker, ReduceMeanChecker, \
26+
ReduceSumChecker, ReluChecker, RequantShiftChecker, ReshapeChecker, RQIntegerDivChecker, SliceChecker, \
27+
SoftmaxChecker, TransposeChecker
2828

2929
BasicTransformer = CodeTransformation([ArgumentStructGeneration(), MemoryManagementGeneration(), FutureGeneration()])
3030

@@ -296,14 +296,14 @@
296296

297297
BasicConvTransposeBindings = [
298298
NodeBinding(
299-
ConvTransposeChecker(
299+
ConvChecker(
300300
[PointerClass(type), PointerClass(type), PointerClass(type)], # input, weight, bias
301301
[PointerClass(type)]),
302302
ConvTransposeTemplate.referenceTemplate,
303303
BasicTransformer) for type in FloatDataTypes
304304
] + [
305305
NodeBinding(
306-
ConvTransposeChecker(
306+
ConvChecker(
307307
[PointerClass(type), PointerClass(type)], # input, weight
308308
[PointerClass(type)]),
309309
ConvTransposeTemplate.referenceTemplate,

Deeploy/Targets/Generic/TypeCheckers.py

Lines changed: 0 additions & 20 deletions
Original file line numberDiff line numberDiff line change
@@ -610,23 +610,3 @@ def _inferNumLevels(self, inputs: List[VariableBuffer],
610610
def _inferSignedness(self, inputs: List[VariableBuffer],
611611
operatorRepresentation: OperatorRepresentation) -> List[bool]:
612612
return [True]
613-
614-
615-
class ConvTransposeChecker(SignPropTypeChecker):
616-
617-
def __init__(self, input_types: Sequence[Type[Pointer]], output_types: Sequence[Type[Pointer]]):
618-
super().__init__(input_types, output_types)
619-
620-
def _inferNumLevels(self, inputs: List[VariableBuffer],
621-
operatorRepresentation: OperatorRepresentation) -> List[int]:
622-
# Like ConvChecker: n_levels = kernel_size * weight_levels * ch_in * 2^bit_input
623-
weight = inputs[1]
624-
return [
625-
np.prod(operatorRepresentation['kernel_shape']) * weight.nLevels * weight.shape[1] * # ch_im_in
626-
2**(self.input_types[0].referencedType.typeWidth)
627-
]
628-
629-
def _inferSignedness(self, inputs: List[VariableBuffer],
630-
operatorRepresentation: OperatorRepresentation) -> List[bool]:
631-
# Output is signed if input is signed
632-
return [inputs[0]._signed]

0 commit comments

Comments
 (0)