forked from jax-ml/jax
-
Notifications
You must be signed in to change notification settings - Fork 4
CI: 03/26/25 upstream sync #316
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
Closed
Closed
Conversation
This file contains hidden or 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
Co-authored-by: Dougal Maclaurin <[email protected]>
PiperOrigin-RevId: 738315605
PiperOrigin-RevId: 738321801
PiperOrigin-RevId: 738342517
http://github.com/openxla/xla/commit/0d20d73f2c8f21c21b9f343c4363a76e980f032e. PiperOrigin-RevId: 738352930
PiperOrigin-RevId: 738376533
…ady `jit`ted. PiperOrigin-RevId: 738393973
PiperOrigin-RevId: 738398099
This code is Mosaic specific, move it to the Mosaic directory. PiperOrigin-RevId: 738404429
PiperOrigin-RevId: 738410122
…MAOp`. Now that we have full control over strides in the lowering, these attributes are no longer necessary. PiperOrigin-RevId: 738418852
…oc around PiperOrigin-RevId: 738421256
This callback functionality is only used by JAX and shipped as part of its CUDA and ROCM GPU plugins. Move it into JAX, as part of a wider move of xla/python pieces that belong to JAX into JAX. PiperOrigin-RevId: 738426489
This is a GPU-specific target. PiperOrigin-RevId: 738441625
PiperOrigin-RevId: 738443014
…xis_index` PiperOrigin-RevId: 738448436
PiperOrigin-RevId: 738454875
PiperOrigin-RevId: 738457532
PiperOrigin-RevId: 738492394
PiperOrigin-RevId: 738496186
PiperOrigin-RevId: 738498184
…e more cleanups PiperOrigin-RevId: 738503430
…ne and WG thread semantics PiperOrigin-RevId: 740371195
PiperOrigin-RevId: 740371651
PiperOrigin-RevId: 740379562
…_mode6 PiperOrigin-RevId: 740381115
PiperOrigin-RevId: 740416272
…allas_1 PiperOrigin-RevId: 740417766
PiperOrigin-RevId: 740422195
PiperOrigin-RevId: 740428623
Introduces a new `download-jax-only-from-gcs` variable to the workflow configs. When set to 1, the test workflows will only download and install the `jax` wheel. Other artifacts such as the latest releases of `jaxlib` and the CUDA plugin dependencies will be downloaded and installed from PyPI. PiperOrigin-RevId: 740430538
PiperOrigin-RevId: 740464973
PiperOrigin-RevId: 740466565
PiperOrigin-RevId: 740478728
It leads to much more compact kernel definitions, just look at the diff! The combination of `pl.core_map` and `pl.run_state` is too noisy to easily follow the kernel logic. PiperOrigin-RevId: 740479934
PiperOrigin-RevId: 740482711
PiperOrigin-RevId: 740487903
PiperOrigin-RevId: 740492139
PiperOrigin-RevId: 740492876
PiperOrigin-RevId: 740505750
PiperOrigin-RevId: 740507886
Drop into `Auto` mode inside for implementation. Co-authored-by: Roy Frostig <[email protected]> PiperOrigin-RevId: 740529498
… in abstract_eval rules of those primitives. PiperOrigin-RevId: 740536031
Drop into `Auto` mode inside for implementation. Co-authored-by: Roy Frostig <[email protected]> PiperOrigin-RevId: 740538785
PiperOrigin-RevId: 740547947
…llbacks instead of failing silently. We will be adding subbyte type support in subsequence changes. PiperOrigin-RevId: 740569954
…llbacks instead of failing silently. We will be adding subbyte type support in subsequence changes. PiperOrigin-RevId: 740577676
PiperOrigin-RevId: 740617998
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