Skip to content

[pass] Remove unused initialized inputs in DCE #2212

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Merged
merged 30 commits into from
Apr 23, 2025
Merged
Changes from 2 commits
Commits
Show all changes
30 commits
Select commit Hold shift + click to select a range
7a720bb
Update unused_removal.py
leshabirukov Apr 17, 2025
62458de
Update unused_removal.py
leshabirukov Apr 17, 2025
dced0cf
Update unused_removal.py
leshabirukov Apr 17, 2025
51cac0d
Update unused_removal.py
leshabirukov Apr 17, 2025
d20287b
Update onnxscript/ir/passes/common/unused_removal.py
leshabirukov Apr 17, 2025
fc1d607
Merge branch 'main' into patch-1
leshabirukov Apr 18, 2025
f6c3eb7
Update unused_removal.py
leshabirukov Apr 18, 2025
d0cafdf
Update __init__.py
leshabirukov Apr 18, 2025
e07b248
Update unused_removal_test.py
leshabirukov Apr 18, 2025
14c918e
Update onnxscript/ir/passes/common/unused_removal.py
leshabirukov Apr 21, 2025
30ad8bd
Update unused_removal.py
leshabirukov Apr 21, 2025
b3f9cde
Update __init__.py
leshabirukov Apr 21, 2025
b00036c
Update unused_removal_test.py
leshabirukov Apr 21, 2025
817709e
Update onnxscript/ir/passes/common/unused_removal_test.py
leshabirukov Apr 22, 2025
716750c
Update unused_removal_test.py
leshabirukov Apr 22, 2025
03ddd34
Update unused_removal.py
leshabirukov Apr 22, 2025
4b138b0
linter. hate it
leshabirukov Apr 22, 2025
f5ff16c
Update unused_removal_test.py
leshabirukov Apr 22, 2025
1b86d37
Update unused_removal_test.py
leshabirukov Apr 22, 2025
a434f3a
Update onnxscript/ir/passes/common/unused_removal_test.py
leshabirukov Apr 22, 2025
e34de72
Update unused_removal_test.py
leshabirukov Apr 22, 2025
39d6979
Update onnxscript/ir/passes/common/unused_removal_test.py
leshabirukov Apr 22, 2025
8407a90
Update onnxscript/ir/passes/common/unused_removal_test.py
leshabirukov Apr 22, 2025
eb0c8ef
Update onnxscript/ir/passes/common/unused_removal_test.py
leshabirukov Apr 22, 2025
dfa809f
Update onnxscript/ir/passes/common/unused_removal_test.py
leshabirukov Apr 22, 2025
84b12cf
Update onnxscript/ir/passes/common/unused_removal_test.py
leshabirukov Apr 22, 2025
6b689ef
Update unused_removal_test.py
leshabirukov Apr 22, 2025
19cf770
Merge branch 'main' into patch-1
leshabirukov Apr 22, 2025
866fbdc
Update unused_removal_test.py
leshabirukov Apr 22, 2025
d12b6b2
Update unused_removal.py
leshabirukov Apr 22, 2025
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
5 changes: 5 additions & 0 deletions onnxscript/ir/passes/common/unused_removal.py
Original file line number Diff line number Diff line change
Expand Up @@ -97,6 +97,11 @@
count = _remove_unused_nodes_in_graph_like(model.graph)
graph_outputs = frozenset(model.graph.outputs)
initializers = model.graph.initializers
graph_inputs = model.graph.inputs
for num, input in list( enumerate( graph_inputs ) )[::-1]:
if input.name in initializers and not (input in graph_outputs or input.uses()):
del graph_inputs[num]
count += 1

Check warning on line 104 in onnxscript/ir/passes/common/unused_removal.py

View check run for this annotation

Codecov / codecov/patch

onnxscript/ir/passes/common/unused_removal.py#L103-L104

Added lines #L103 - L104 were not covered by tests
for init in list(initializers.values()):
if not (init in graph_outputs or init.uses()):
assert init.name is not None
Expand Down
Loading