8000 [ONNX] Add support for torch.cond/HOP in onnx exporter by xadupre · Pull Request #137428 · pytorch/pytorch · GitHub
[go: up one dir, main page]
Skip to content

[ONNX] Add support for torch.cond/HOP in onnx exporter #137428

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.

Sign up for GitHub

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

Closed
wants to merge 47 commits into from
Closed

[ONNX] Add support for torch.cond/HOP in onnx exporter #137428

Changes from 1 commit
Commits
Show all changes
47 commits
Select commit Hold shift + click to select a range
d1cc835
[ONNX] Support model with torch.cond
xadupre Oct 7, 2024
3d89910
fix annotation
xadupre Oct 7, 2024
e4b5870
Merge branch 'main' of https://github.com/pytorch/pytorch into onnx_cond
xadupre Nov 4, 2024
de5790a
complete support of torch.cond
xadupre Nov 4, 2024
ede7891
Merge branch 'main' of https://github.com/pytorch/pytorch into onnx_cond
xadupre Nov 5, 2024
9ec86b7
use ir
xadupre Nov 5, 2024
a1627cf
lint
xadupre Nov 5, 2024
098fa44
Update test_small_models_e2e.py
justinchuby Nov 17, 2024
8b4dc0c
Merge branch 'main' of https://github.com/pytorch/pytorch into onnx_cond
xadupre Nov 18, 2024
7eee884
Update _core.py
justinchuby Nov 19, 2024
01a34e1
Update _building.py
justinchuby Nov 19, 2024
d5915e4
Update _fx_passes.py
justinchuby Nov 19, 2024
f613214
Update _ir_passes.py
justinchuby Nov 19, 2024
c45884d
Update test_small_models_e2e.py
justinchuby Nov 19, 2024
6cba4a9
Update _core.py
justinchuby Nov 19, 2024
83bd6bc
Update _registration.py
justinchuby Nov 19, 2024
9e7e95d
Create _torchlib_registry.py
justinchuby Nov 19, 2024
e019ec2
Create __init__.py
justinchuby Nov 19, 2024
8a2b2af
Create hop.py
justinchuby Nov 19, 2024
fc8b09b
Update hop.py
justinchuby Nov 19, 2024
7f7f59a
Update test_small_models_e2e.py
justinchuby Nov 19, 2024
70d89fe
Update _core.py
justinchuby Nov 19, 2024
5756204
Update _ir_passes.py
justinchuby Nov 19, 2024
1134b13
Update _building.py
justinchuby Nov 19, 2024
c8c0194
Update _schemas.py
justinchuby Nov 19, 2024
94ff4e3
Update _ir_passes.py
justinchuby Nov 19, 2024
e8b8e86
Update _core.py
justinchuby Nov 19, 2024
668072e
Update _core.py
justinchuby Nov 19, 2024
d232470
Create __init__.py
justinchuby Nov 19, 2024
c4771ac
Update hop.py
justinchuby Nov 19, 2024
fe9a778
Update _core.py
justinchuby Nov 19, 2024
c14013e
Update test_small_models_e2e.py
justinchuby Nov 19, 2024
ff9c2f8
Update test_small_models_e2e.py
justinchuby Nov 19, 2024
d6ccefa
Update hop.py
justinchuby Nov 19, 2024
3ac734f
Update hop.py
justinchuby Nov 19, 2024
ca634a1
Update test_small_models_e2e.py
justinchuby Nov 19, 2024
f04378c
Update _building.py
justinchuby Nov 19, 2024
d4c8447
Update _core.py
justinchuby Nov 19, 2024
600e23e
Update _fx_passes.py
justinchuby Nov 19, 2024
d1869c5
Update _registration.py
justinchuby Nov 19, 2024
9616504
Update _schemas.py
justinchuby Nov 19, 2024
7d2c89a
Update _torchlib_registry.py
justinchuby Nov 19, 2024
f9165b5
Update _building.py
justinchuby Nov 19, 2024
fd86370
Update _registration.py
justinchuby Nov 19, 2024
00b883d
Update test_small_models_e2e.py
justinchuby Nov 20, 2024
eda5c04
Update __init__.py
justinchuby Nov 20, 2024
9f081b8
Update _ir_passes.py
justinchuby Nov 20, 2024
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
Prev Previous commit
Update _ir_passes.py
  • Loading branch information
@justinchuby
justinchuby authored Nov 20, 2024
commit 9f081b815411c321c3b1511c9382583d849f1370
5 changes: 5 additions & 0 deletions torch/onnx/_internal/exporter/_ir_passes.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -68,3 +68,8 @@ def add_opset_imports(model: ir.Model) -> None:
function.opset_imports[domain] = _get_opset_version(
node, function.opset_imports
)
for opset, version in function.opset_imports.items():
# Add all opsets used in the function to the model, because ONNX Runtime
# does not handle adding the opset imports to the model after inlining during inference.
# This should happen after all opsets are collected for the function from its nodes.
model.opset_imports[opset] = max(version, model.opset_imports.get(opset, 1))
Loading
0