Skip to content

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
wants to merge 193 commits into from

Conversation

rocm-repo-management-api-2[bot]
Copy link

Daily sync with upstream

charleshofer and others added 30 commits March 11, 2025 14:57
This code is Mosaic specific, move it to the Mosaic directory.

PiperOrigin-RevId: 738404429
…MAOp`.

Now that we have full control over strides in the lowering, these attributes
are no longer necessary.

PiperOrigin-RevId: 738418852
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
…e more cleanups

PiperOrigin-RevId: 738503430
nvcastet and others added 28 commits March 25, 2025 10:20
…ne and WG thread semantics

PiperOrigin-RevId: 740371195
PiperOrigin-RevId: 740371651
PiperOrigin-RevId: 740379562
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
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: 740505750
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
…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
@rocm-repo-management-api-2 rocm-repo-management-api-2 bot requested a review from a team as a code owner March 26, 2025 06:02
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

Successfully merging this pull request may close these issues.