8000 [scan] Autograd with partial gradient support by bohnstingl · Pull Request #146285 · pytorch/pytorch · GitHub
[go: up one dir, main page]
Skip to content

[scan] Autograd with partial gradient support #146285

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 54 commits into from
Closed

[scan] Autograd with partial gradient support #146285

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
54 commits
Select commit Hold shift + click to select a range
f6586dd
WIP: Integration of scan autograd
bohnstingl Feb 1, 2025
bda8cb9
Introduced autograd for scan with partial gradients support
bohnstingl Feb 3, 2025
34be923
Fixed CI test issue with graph
bohnstingl Feb 6, 2025
5eadb1e
Merge branch 'main' of github.com:pytorch/pytorch into scan_autograd22
bohnstingl Feb 7, 2025
adfa593
Integrated code review comments
bohnstingl Feb 18, 2025
7b41bde
Merge branch 'main' of github.com:pytorch/pytorch into scan_autograd22
bohnstingl Feb 19, 2025
c2465cf
Fixed type annotation
bohnstingl Feb 20, 2025
5bc1d27
Removed reverse flag from backend and implemented reverse with torch.…
bohnstingl Feb 25, 2025
2786a68
Fix of graph in testcase
bohnstingl Feb 25, 2025
7f48b4f
Merge branch 'scan_flip_reverse' of github.com:bohnstingl/pytorch int…
bohnstingl Feb 26, 2025
6717f9b
Integrated new reverse handling into scan autograd
bohnstingl Feb 26, 2025
a60d2a8
Merge branch 'main' of github.com:pytorch/pytorch into scan_autograd22
bohnstingl Mar 7, 2025
0343056
Fixed issues with testcases and with combine_fn now returning pytrees
bohnstingl Mar 7, 2025
0dda954
[cond] don't trace fw and bw graph in autograd key
ydwu4 Mar 10, 2025
c11cac4
Update on "[cond] don't trace fw and bw graph in autograd key"
ydwu4 Mar 17, 2025
ebb5b57
Update on "[cond] don't trace fw and bw graph in autograd key"
ydwu4 Mar 18, 2025
46608f8
Update on "[cond] don't trace fw and bw graph in autograd key"
ydwu4 Mar 18, 2025
26e4a53
Update on "[cond] don't trace fw and bw graph in autograd key"
ydwu4 Mar 19, 2025
813c883
Merge branch 'main' of github.com:pytorch/pytorch into scan_autograd22
bohnstingl Mar 19, 2025
9e4f3c3
Merge branch 'gh/ydwu4/222/head' of github.com:pytorch/pytorch into s…
bohnstingl Mar 19, 2025
114c982
Update base for Update on "[cond] don't trace fw and bw graph in auto…
ydwu4 Mar 20, 2025
fdbfe11
Update on "[cond] don't trace fw and bw graph in autograd key"
ydwu4 Mar 20, 2025
7796ea9
Update base for Update on "[cond] don't trace fw and bw graph in auto…
ydwu4 Mar 20, 2025
6082a60
Update on "[cond] don't trace fw and bw graph in autograd key"
ydwu4 Mar 20, 2025
d33705f
Update base for Update on "[cond] don't trace fw and bw graph in auto…
ydwu4 Mar 20, 2025
7cee41c
Update on "[cond] don't trace fw and bw graph in autograd key"
ydwu4 Mar 20, 2025
23170b1
WIP: scan autograd
bohnstingl Mar 21, 2025
0ac5a15
Merge branch 'gh/ydwu4/222/head' of github.com:pytorch/pytorch into s…
bohnstingl Mar 21, 2025
93f5bab
Update base for Update on "[cond] don't trace fw and bw graph in auto…
ydwu4 Mar 21, 2025
ad92afa
Update on "[cond] don't trace fw and bw graph in autograd key"
ydwu4 Mar 21, 2025
5947348
Merge branch 'gh/ydwu4/222/head' of github.com:pytorch/pytorch into s…
bohnstingl Mar 22, 2025
60fc08f
Working function of scan autograd with new infrastructure
bohnstingl Mar 24, 2025
75fe416
Removed unnecessary code and other cleanups
bohnstingl Mar 24, 2025
26c94e9
Merge branch 'main' of github.com:pytorch/pytorch into scan_autograd22
bohnstingl Mar 24, 2025
357dde5
Merge branch 'main' of github.com:pytorch/pytorch into scan_autograd22
bohnstingl Mar 26, 2025
b018397
Fixed tracing issue causing missing fake meta values
bohnstingl Mar 27, 2025
8a288d3
Merge branch 'main' of github.com:pytorch/pytorch into scan_autograd22
bohnstingl Mar 27, 2025
63414f9
Fixed CI test
bohnstingl Mar 28, 2025
66f0b37
Merge branch 'main' of github.com:pytorch/pytorch into scan_autograd22
bohnstingl Mar 28, 2025
b50e7b9
Merge branch 'main' of github.com:pytorch/pytorch into scan_autograd22
bohnstingl Mar 31, 2025
0db488f
Fixed testcase
bohnstingl Mar 31, 2025
9df0724
Integrated review feedback; before code cleanup
bohnstingl Apr 1, 2025
ffacadd
After cleanup and lintrunner
bohnstingl Apr 2, 2025
4ce5f06
Fixes
bohnstingl Apr 2, 2025
88084f5
Further cleanup and restructuring
bohnstingl Apr 2, 2025
473db76
WIP: integration of code review
bohnstingl Apr 3, 2025
e6ca8b0
Added extensive documentation and lintrunner corrections
bohnstingl Apr 3, 2025
d27d50a
Revert change to cond.py
bohnstingl Apr 3, 2025
0ee4d01
Removed unrelated testcases and fixed lint issues
bohnstingl Apr 3, 2025
66c0ae1
Integrated review changes for documentation and test cases
bohnstingl Apr 4, 2025
c976b5e
Added some more comments
bohnstingl Apr 8, 2025
afa5d96
Introduced materialize_as_graph for combine_fn and for combine_bw_fn
bohnstingl Apr 9, 2025
6d67d3e
Merge branch 'main' of github.com:pytorch/pytorch into scan_autograd22
bohnstingl Apr 10, 2025
ebe8c22
Added bypass for autograd DispatchKey if no gradient for either the i…
bohnstingl Apr 10, 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
Loading
Loading
0