|
9 | 9 | from Deeploy.CommonExtensions.CodeTransformationPasses.Closure import ClosureGeneration, MemoryAwareClosureGeneration |
10 | 10 | from Deeploy.CommonExtensions.CodeTransformationPasses.MemoryAllocation import ArgumentStructGeneration, \ |
11 | 11 | MemoryManagementGeneration, MemoryPassthroughGeneration |
12 | | -from Deeploy.CommonExtensions.DataTypes import IntegerDataTypes, SignedIntegerDataTypes, float32_t, int8_t, int32_t, \ |
13 | | - uint8_t |
| 12 | +from Deeploy.CommonExtensions.DataTypes import FloatDataTypes, IntegerDataTypes, SignedIntegerDataTypes, float32_t, \ |
| 13 | + int8_t, int32_t, int64_t, uint8_t |
14 | 14 | from Deeploy.DeeployTypes import CodeTransformation, NodeBinding, NodeTemplate |
15 | 15 | from Deeploy.FutureExtension.Bindings.AutoFutureBinding import AutoFutureBinding |
16 | 16 | from Deeploy.FutureExtension.CodeTransformationPasses.FutureCodeTransformation import FutureGeneration |
|
153 | 153 | ] |
154 | 154 |
|
155 | 155 | PULPReshapeBindings = [ |
156 | | - NodeBinding(ReshapeChecker([PointerClass(type), PointerClass(int32_t)], [PointerClass(type)]), |
157 | | - ReshapeTemplate.referenceTemplate, SkipTransformer) for type in IntegerDataTypes |
158 | | -] + [ |
159 | | - NodeBinding(ReshapeChecker([PointerClass(float32_t), PointerClass(type)], [PointerClass(float32_t)]), |
160 | | - ReshapeTemplate.referenceTemplate, SkipTransformer) for type in IntegerDataTypes |
| 156 | + NodeBinding(ReshapeChecker([PointerClass(type), PointerClass(int64_t)], [PointerClass(type)]), |
| 157 | + ReshapeTemplate.referenceTemplate, SkipTransformer) for type in IntegerDataTypes + FloatDataTypes |
161 | 158 | ] |
162 | 159 |
|
163 | 160 | PULPRQAddBindings = [ |
|
0 commit comments