pytorch-lightning
Allow to return ONNXProgram when calling `to_onnx(dynamo=True)`
#20811
Merged
Merged
Allow to return ONNXProgram when calling `to_onnx(dynamo=True)` #20811
Borda
merged
49 commits into
Lightning-AI:master
from
GdoongMathew:feat/dynamo_export_onnx
feat: return `ONNXProgram` when exporting with dynamo=True.
bde96143
test: add to_onnx(dynamo=True) unittests.
a966c0b5
fix: add ignore filter in pyproject.toml
e7342e38
deependujha
commented
on 2025-05-20
deependujha
commented
on 2025-05-20
deependujha
approved these changes
on 2025-05-20
fix: change the return type annotation of `to_onnx`.
3ee3ea92
test: add parametrized `dynamo` to test `test_if_inference_output_is_…
bc812155
test: add difference check in `test_model_return_type`.
236f1a07
fix: fix unittest.
019125d5
GdoongMathew
changed the title
Merge branch 'master' into feat/dynamo_export_onnx
9f5e6049
deps: bump typing_extension for onnxscript.
791d7773
Merge branch 'master' into feat/dynamo_export_onnx
453e63fc
deps: bump typing_extension for onnxscript.
acdf3c12
deps: bump onnxscript upper bound.
e046d272
test: add test `test_model_onnx_export_missing_onnxscript`.
a0a7d1f2
Borda
approved these changes
on 2025-06-06
Merge branch 'master' into feat/dynamo_export_onnx
7aae865c
Borda
commented
on 2025-06-06
revert typing-extension bump.
aca9fd1c
lower the min_torch version in unittest.
1396f353
feat: enable ONNXProgram export on torch 2.5.0
8f050ea5
Merge branch 'master' into feat/dynamo_export_onnx
3938c735
extensions
ce3e6b74
Merge branch 'master' into feat/dynamo_export_onnx
c31a3f6b
Merge branch 'master' into feat/dynamo_export_onnx
40b1449e
ds
a470fe85
![pre-commit-ci[bot]](https://avatars.githubusercontent.com/in/68672?v=4&s=60)
[pre-commit.ci] auto fixes from pre-commit.com hooks
9e4a4947
Merge branch 'master' into feat/dynamo_export_onnx
b08e4650
dep: test fixing pydantic version.
67af423b
Revert "dep: test fixing pydantic version."
0e4cb807
dep: add serve deps.
b26072d7
ci: test.
d1b85976
GdoongMathew
commented
on 2025-06-18
Merge branch 'master' into feat/dynamo_export_onnx
ec638c30
update onnxscript upperbound.
aa951fd6
align with ce3e6b7
9491953f
Merge branch 'master' into feat/dynamo_export_onnx
819d3c8f
Merge branch 'master' into feat/dynamo_export_onnx
f0146394
Merge branch 'master' into feat/dynamo_export_onnx
784fc3f2
Merge branch 'master' into feat/dynamo_export_onnx
12570c14
Borda
changed the title
Borda
changed the title
Borda
approved these changes
on 2025-08-08
chlog
7c2bfee4
Borda
commented
on 2025-08-08
drop serve
ee596a6c
import
caba43cb
prune
23e5a1c3
Apply suggestions from code review
e8595f81
req
87d8f179
prune
eeb1a136
import
87bdfd91
[pre-commit.ci] auto fixes from pre-commit.com hooks
3d9ecc87
import
0493e13b
Merge branch 'feat/dynamo_export_onnx' of https://github.com/GdoongMa…
d6a1e8e6
import
bdf6927f
ignore[no-redef]
1fd6e76a
[pre-commit.ci] auto fixes from pre-commit.com hooks
ee190854
Reviewers
Assignees
No one assigned
Labels
docs
ci
fabric
pl
dependencies
package
Milestone
No milestone
Login to write a write a comment.
Login via GitHub