forked from jax-ml/jax
-
Notifications
You must be signed in to change notification settings - Fork 4
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
CI: 01/07/25 upstream sync #194
Merged
Merged
Conversation
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
PiperOrigin-RevId: 712463302
…scipy.special.sph_harm`. These functions are deprecated in scipy 1.15.0. I'll fix this properly soon, but let's start by getting CI working again! PiperOrigin-RevId: 712512363
… 13. PiperOrigin-RevId: 712516025
This was deprecated in January 2024, replaced by `core_max_dim(..., 0)`. PiperOrigin-RevId: 712523579
PiperOrigin-RevId: 712544847
http://github.com/openxla/xla/commit/1a6361a734c5cd10dc93938fc6163a51fd37b82e. PiperOrigin-RevId: 712558157
PiperOrigin-RevId: 712579230
PiperOrigin-RevId: 712587802
PiperOrigin-RevId: 712594991
PiperOrigin-RevId: 712595778
PiperOrigin-RevId: 712603103
In our MSAN CI, the copy of LAPACK we use is not MSAN-instrumented, leading to false positives. Suppress those false-positives via annotations. PiperOrigin-RevId: 712607044
We have no remaining uses of pybind11 outside a GPU custom call example. PiperOrigin-RevId: 712608834
PiperOrigin-RevId: 712630197
PiperOrigin-RevId: 712671502
PiperOrigin-RevId: 712689888
PiperOrigin-RevId: 712702329
PiperOrigin-RevId: 712707311
Previously, we could only handle threefry for the case when it was possible to tell statically that the size of the `count` array is even or odd. This meant that often we had to add a constraint that one of the dimensions is even. Here we rewrite the handling of threefry to not require a Python-level conditional about evenness of the size of the count array. We use a couple of `lax.dynamic_slice` rather than a `lax.split`. We also generalize the tests to cases where the size if fully symbolic, and we cannot tell statically that it is even.
PiperOrigin-RevId: 712826758
PiperOrigin-RevId: 712863350
PiperOrigin-RevId: 712864349
This enables dynamic registration of custom FFI types on the appropriate platform via PJRT. PiperOrigin-RevId: 712904085
charleshofer
approved these changes
Jan 7, 2025
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
Add this suggestion to a batch that can be applied as a single commit.
This suggestion is invalid because no changes were made to the code.
Suggestions cannot be applied while the pull request is closed.
Suggestions cannot be applied while viewing a subset of changes.
Only one suggestion per line can be applied in a batch.
Add this suggestion to a batch that can be applied as a single commit.
Applying suggestions on deleted lines is not supported.
You must change the existing code in this line in order to create a valid suggestion.
Outdated suggestions cannot be applied.
This suggestion has been applied or marked resolved.
Suggestions cannot be applied from pending reviews.
Suggestions cannot be applied on multi-line comments.
Suggestions cannot be applied while the pull request is queued to merge.
Suggestion cannot be applied right now. Please check back later.
Daily sync with upstream