Skip to content

Commit abdfaba

Browse files
committed
Fix CI
1 parent a4b25e7 commit abdfaba

File tree

1 file changed

+5
-4
lines changed

1 file changed

+5
-4
lines changed

DeeployTest/testMVP.py

Lines changed: 5 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -40,6 +40,7 @@
4040
from testUtils.testRunner import TestGeneratorArgumentParser
4141
from testUtils.typeMapping import inferInputType
4242

43+
from Deeploy.AbstractDataTypes import PointerClass
4344
from Deeploy.DeeployTypes import CodeGenVerbosity, ConstantBuffer, NetworkContext, NetworkDeployer, ONNXLayer, \
4445
SubGraph, TransientBuffer
4546
from 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

Comments
 (0)