8000 [associative_scan] Autograd for additional inputs by bohnstingl · Pull Request #153317 · pytorch/pytorch · GitHub
[go: up one dir, main page]
Skip to content

[associative_scan] Autograd for additional inputs #153317

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

Jump to bottom
Draft
wants to merge 34 commits into
base: main
Choose a base branch
from
Draft

[associative_scan] Autograd for additional inputs #153317

wants to merge 34 commits into from

Conversation

bohnstingl
Copy link
Collaborator
@bohnstingl bohnstingl commented May 10, 2025 •
edited by pytorch-bot bot
Loading

@bohnstingl
WIP: Associative_scan Autograd
9c49a36
@bohnstingl
Working implementation of Autograd
100c598
@bohnstingl
Added partial gradient tests
0e7c8d5
@bohnstingl
Separated out the partial gradient functionality to a separate PR
67d62fb
@bohnstingl
Merge branch 'main' of github.com:pytorch/pytorch into associative_sc…
9e5e1f9 
…an_74
@bohnstingl
Working version uncleaned
6b41565
@bohnstingl
Almost all tests pass
d68b31b
@bohnstingl
Merge branch 'main' of github.com:pytorch/pytorch into associative_sc…
9de0caf 
…an_74
@bohnstingl
First working implementation of simplified autograd for combine_mode=…
653eab0 
…pointwise
@bohnstingl
Merge branch 'associative_scan_74' of github.com:bohnstingl/pytorch i…
ac1a12b 
…nto associative_scan_74
@bohnstingl
Merge branch 'main' of github.com:pytorch/pytorch into associative_sc…
d51b16c 
…an_74
@bohnstingl
WIP: No additional_input support yet
dbfd544
@bohnstingl
Updates and cosmetic fixes
64a8b29
@bohnstingl
iFixed problem with python<3.11
08b7251
@bohnstingl
Merge branch 'main' of github.com:pytorch/pytorch into associative_sc…
f378827 
…an_74
@bohnstingl
Fixed issue with adding tuple and list
b12777e
@bohnstingl
Merge branch 'main' of github.com:pytorch/pytorch into associative_sc…
6335022 
…an_74
@bohnstingl
Prepared backward of associative_scan for additional_inputs
89ea84c
@bohnstingl
Cleaned-up code
c8bbdf4
@bohnstingl
Merge branch 'main' of github.com:pytorch/pytorch into associative_sc…
1c9fa70 
…an_74
@bohnstingl
Merge branch 'main' of github.com:pytorch/pytorch into associative_sc…
6d8353b 
…an_74
@bohnstingl
Rework to improve readability and unify shared function with scan
3f81639
@bohnstingl
Removed irrelevant testcases, improved readability, extended document…
62770d5 
…ation and fixed lint issues
@bohnstingl
Restructured documentation and synced with associative_scan
6761714
@bohnstingl
Merge branch 'main' of github.com:pytorch/pytorch into associative_sc…
b73c553 
…an_74
@bohnstingl
Merge branch 'main' of github.com:pytorch/pytorch into associative_sc…
ea9568d 
…an_74
@bohnstingl
Merge branch 'main' of github.com:pytorch/pytorch into associative_sc…
0786e6f 
…an_74
@bohnstingl
Merge branch 'main' of github.com:pytorch/pytorch into associative_sc…
c628dd8 
…an_74
@bohnstingl
Skipping autograd test for associative_scan with lifted arguments
e7d63b6
@bohnstingl
Reworked documentation
95d0e42
@pytorch-bot PyTorch Bot
Copy link
pytorch-bot bot commented May 10, 2025 •
edited
Loading

🔗 Helpful Links

🧪 See artifacts and rendered test results at hud.pytorch.org/pr/153317

Note: Links to docs will display an error until the docs builds have been completed.

❌ 2 New Failures, 1 Cancelled Job, 2 Unrelated Failures

As of commit 2edd60d with merge base dd7d231 (image):

NEW FAILURES - The following jobs have failed:

CANCELLED JOB - The following job was cancelled. Please retry:

UNSTABLE - The following jobs are marked as unstable, possibly due to flakiness on trunk:

This comment was automatically generated by Dr. CI and updates every 15 minutes.

@bohnstingl
Copy link
Collaborator Author

@pytorchbot label "topic: not user facing"

@pytorch-bot pytorch-bot bot added the topic: not user facing topic category label May 10, 2025
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Reviewers

@zou3519 zou3519 Awaiting requested review from zou3519 zou3519 will be requested when the pull request is marked ready for review zou3519 is a code owner

Assignees
No one assigned
Labels
ciflow/inductor module: dynamo open source topic: not user facing topic category
Projects
None yet
Milestone
No milestone
Development

Successfully merging this pull request may close these issues.
2 participants
@bohnstingl @pytorchbot
0