4040from testUtils .testRunner import TestGeneratorArgumentParser
4141from testUtils .typeMapping import inferInputType
4242
43+ from Deeploy .AbstractDataTypes import PointerClass
4344from Deeploy .DeeployTypes import CodeGenVerbosity , ConstantBuffer , NetworkContext , NetworkDeployer , ONNXLayer , \
4445 SubGraph , TransientBuffer
4546from Deeploy .EngineExtension .NetworkDeployers .EngineColoringDeployer import EngineColoringDeployerWrapper
@@ -148,7 +149,6 @@ def setupDeployer(graph: gs.Graph, memoryHierarchy: MemoryHierarchy, defaultTarg
148149 _DEEPLOYSTATEDIR = os .path .join (args .dumpdir , "deeployStates" )
149150
150151 inputs = np .load (f'{ args .dir } /inputs.npz' )
151- tensors = graph .tensors ()
152152
153153 # Load as int64 and infer types later
154154 test_inputs = [inputs [x ].reshape (- 1 ).astype (np .float64 ) for x in inputs .files ]
@@ -160,12 +160,13 @@ def setupDeployer(graph: gs.Graph, memoryHierarchy: MemoryHierarchy, defaultTarg
160160 if args .enableStrides :
161161 platform .engines [0 ].enableStrides = True
162162
163- for index , num in enumerate (test_inputs ):
164- # WIESP: Do not infer types and offset of empty arrays
163+ for index , (name , num ) in enumerate (zip (inputs .files , test_inputs )):
165164 if np .prod (num .shape ) == 0 :
166165 continue
166+
167167 _type , offset = inferInputType (num , signProp )[0 ]
168- inputTypes [f"input_{ index } " ] = _type
168+
169+ inputTypes [f"input_{ index } " ] = PointerClass (_type )
169170 inputOffsets [f"input_{ index } " ] = offset
170171
171172 deployer = mapDeployer (platform ,
0 commit comments