8000 [associative_scan] scan dim handling in user-facing associative_scan() by bohnstingl · Pull Request #139864 · pytorch/pytorch · GitHub
[go: up one dir, main page]
Skip to content

[associative_scan] scan dim handling in user-facing associative_scan() #139864

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

[associative_scan] scan dim handling in user-facing associative_scan() #139864

Changes from 1 commit
Commits
Show all changes
22 commits
Select commit Hold shift + click to select a range
460f753
Ensure that the combine_fn is only called with the proper slice of th…
bohnstingl Oct 24, 2024
1419a79
Fixed shape check
bohnstingl Oct 24, 2024
944649a
WIP: nested associative_scan
bohnstingl Oct 24, 2024
f974cf3
Incorporated first review round
bohnstingl Oct 26, 2024
ab0e515
Implemented better and more unified testing procedures
bohnstingl Oct 26, 2024
59b164b
Rebase to main
bohnstingl Oct 26, 2024
6dc7811
Lintrunner cleanup
bohnstingl Oct 26, 2024
308e89c
WIP: new _run_test interface
bohnstingl Oct 29, 2024
0a902eb
Integrated comments from PR and updated testcases
bohnstingl Oct 30, 2024
022a454
Integrated nested tuple for the vmap used in generic_associative_scan
bohnstingl Oct 30, 2024
8aeef66
Integrated nit changes
bohnstingl Oct 31, 2024
9e01fff
Fixed minor issue with testcase parameters
bohnstingl Oct 31, 2024
90e9ac3
Rebased to associative_scan_70
bohnstingl Oct 31, 2024
ce619ea
Fixed rebasing issues
bohnstingl Oct 31, 2024
85a703c
Merge branch 'main' of github.com:pytorch/pytorch into associative_sc…
bohnstingl Nov 6, 2024
5564064
Merge branch 'main' of github.com:pytorch/pytorch into associative_sc…
bohnstingl Jan 18, 2025
56b299d
Fixed merge conflicts
bohnstingl Jan 18, 2025
f82f6b0
Corrections for lintrunner
bohnstingl Jan 18, 2025
c147eec
Created generic dim moving function that can be reused
bohnstingl Jan 19, 2025
845d366
Updated assertions
bohnstingl Jan 19, 2025
1921ab0
Merge branch 'main' of github.com:pytorch/pytorch into associative_sc…
bohnstingl Jan 23, 2025
a029df1
Removed wrapper around `shift_source_dim_to_target_dim`
bohnstingl Jan 23, 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
Prev Previous commit
Next Next commit
Integrated comments from PR and updated testcases
  • Loading branch information
@bohnstingl
bohnstingl committed Oct 30, 2024
commit 0a902eb2489a1beac87a389dc1a212448439f94e
Loading
Loading
0