Skip to content
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.

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

[mutable-arrays] move MutableArray, add eager, improve tests, fix bug #20044

Merged
merged 1 commit into from
Mar 1, 2024

Conversation

mattjj
Copy link
Collaborator

@mattjj mattjj commented Mar 1, 2024

  1. move MutableArray to core.py, and some handlers to their respective files
  2. fix a bug in aliasing setup (it was just broken before, now better test coverage)
  3. add eager support by enabling get_p, swap_p, and addupdate_p impls
  4. improve tests slightly

@mattjj mattjj requested review from froystig and sharadmv March 1, 2024 19:13
@mattjj mattjj force-pushed the mutable-arrays branch 2 times, most recently from db28f5e to 3e68277 Compare March 1, 2024 19:26
@google-ml-butler google-ml-butler bot added kokoro:force-run pull ready Ready for copybara import and testing labels Mar 1, 2024
@mattjj mattjj force-pushed the mutable-arrays branch 2 times, most recently from 2a0c5ec to 3a94a4e Compare March 1, 2024 22:26
1. move MutableArray to core.py, and some handlers to their respective files
2. fix a bug in aliasing setup (it was just broken before, now better test coverage)
3. add eager support by enabling get_p, swap_p, and addupdate_p impls
4. improve tests slightly
@copybara-service copybara-service bot merged commit 28f84eb into jax-ml:main Mar 1, 2024
13 checks passed
@mattjj mattjj deleted the mutable-arrays branch March 1, 2024 23:32
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
pull ready Ready for copybara import and testing
Projects
None yet
Development

Successfully merging this pull request may close these issues.

4 participants