Skip to content

Commit 38d7e10

Browse files
committed
fix issues
1 parent a6cfc23 commit 38d7e10

File tree

1 file changed

+7
-4
lines changed

1 file changed

+7
-4
lines changed

onnx_diagnostic/reference/ort_evaluator.py

Lines changed: 7 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -267,7 +267,8 @@ def run(
267267
self._log(2, " + %s: %s", name, value) # type: ignore[arg-type]
268268
assert isinstance(name, str), f"unexpected type for name {type(name)}"
269269
results[name] = value
270-
self._clean_unused_inplace(i_node, node, results)
270+
if not intermediate:
271+
self._clean_unused_inplace(i_node, node, results)
271272

272273
if intermediate:
273274
return results
@@ -314,16 +315,18 @@ def _clean_unused_inplace(self, i_node: int, node: NodeProto, results: Dict[str,
314315
if not self.garbage_collector:
315316
return
316317
for name in node.input:
317-
if self.garbage_collector[name] == i_node:
318+
if self.garbage_collector[name] == i_node and name in results:
318319
if self.verbose:
319-
print(f" - deletes: {name}")
320+
t = results[name]
321+
print(f" - deletes: {name} - {t.dtype}:{t.shape}")
320322
del results[name]
321323
if node.op_type in {"Scan", "If", "Loop"}:
322324
hidden = self._get_hidden_node_inputs(node)
323325
for name in hidden:
324326
if self.garbage_collector[name] == i_node and name in results:
325327
if self.verbose:
326-
print(f" - deletes: {name}")
328+
t = results[name]
329+
print(f" - deletes: {name} - {t.dtype}:{t.shape}")
327330
del results[name]
328331

329332
def _make_model_proto(

0 commit comments

Comments
 (0)