Skip to content

Update rules_python to v1.6 in JAX.#31709

Merged
copybara-service[bot] merged 1 commit intomainfrom
test_805389759
Sep 18, 2025
Merged

Update rules_python to v1.6 in JAX.#31709
copybara-service[bot] merged 1 commit intomainfrom
test_805389759

Conversation

@copybara-service
Copy link
Copy Markdown

@copybara-service copybara-service Bot commented Sep 10, 2025

Update rules_python to v1.6 in JAX.

Also remove stale files from the wheel (see link).

copybara-service Bot pushed a commit to openxla/xla that referenced this pull request Sep 10, 2025
`com_google_protobuf` and `rules_cc` initialization was moved to `python_init_rules` macros.

The JAX repository update will be done [in the next step](jax-ml/jax#31709)

PiperOrigin-RevId: 803189916
copybara-service Bot pushed a commit to openxla/xla that referenced this pull request Sep 10, 2025
`com_google_protobuf` and `rules_cc` initialization was moved to `python_init_rules` macros.

The JAX repository update will be done [in the next step](jax-ml/jax#31709)

PiperOrigin-RevId: 803189916
copybara-service Bot pushed a commit to tensorflow/tensorflow that referenced this pull request Sep 10, 2025
`com_google_protobuf` and `rules_cc` initialization was moved to `python_init_rules` macros.

The JAX repository update will be done [in the next step](jax-ml/jax#31709)

PiperOrigin-RevId: 803189916
copybara-service Bot pushed a commit to openxla/xla that referenced this pull request Sep 15, 2025
`com_google_protobuf` and `rules_cc` initialization was moved to `python_init_rules` macros.

The JAX repository update will be done [in the next step](jax-ml/jax#31709)

PiperOrigin-RevId: 803189916
copybara-service Bot pushed a commit to tensorflow/tensorflow that referenced this pull request Sep 15, 2025
`com_google_protobuf` and `rules_cc` initialization was moved to `python_init_rules` macros.

The JAX repository update will be done [in the next step](jax-ml/jax#31709)

FUTURE_COPYBARA_INTEGRATE_REVIEW=#100135 from ILCSFNO:patch-4 20d77c2
PiperOrigin-RevId: 803189916
copybara-service Bot pushed a commit to openxla/xla that referenced this pull request Sep 16, 2025
`com_google_protobuf` and `rules_cc` initialization was moved to `python_init_rules` macros.

The JAX repository update will be done [in the next step](jax-ml/jax#31709)

PiperOrigin-RevId: 803189916
copybara-service Bot pushed a commit to tensorflow/tensorflow that referenced this pull request Sep 16, 2025
`com_google_protobuf` and `rules_cc` initialization was moved to `python_init_rules` macros.

The JAX repository update will be done [in the next step](jax-ml/jax#31709)

PiperOrigin-RevId: 803189916
copybara-service Bot pushed a commit to openxla/xla that referenced this pull request Sep 16, 2025
`com_google_protobuf` and `rules_cc` initialization was moved to `python_init_rules` macros.

The JAX repository update will be done [in the next step](jax-ml/jax#31709)

PiperOrigin-RevId: 803189916
copybara-service Bot pushed a commit to openxla/xla that referenced this pull request Sep 16, 2025
`com_google_protobuf` and `rules_cc` initialization was moved to `python_init_rules` macros.

The JAX repository update will be done [in the next step](jax-ml/jax#31709)

PiperOrigin-RevId: 803189916
copybara-service Bot pushed a commit to tensorflow/tensorflow that referenced this pull request Sep 16, 2025
`com_google_protobuf` and `rules_cc` initialization was moved to `python_init_rules` macros.

The JAX repository update will be done [in the next step](jax-ml/jax#31709)

PiperOrigin-RevId: 803189916
@copybara-service copybara-service Bot force-pushed the test_805389759 branch 7 times, most recently from 0c08364 to 0fe55ef Compare September 17, 2025 17:35
copybara-service Bot pushed a commit to openxla/xla that referenced this pull request Sep 17, 2025
`com_google_protobuf` and `rules_cc` initialization was moved to `python_init_rules` macros.

The JAX repository update will be done [in the next step](jax-ml/jax#31709)

PiperOrigin-RevId: 803189916
copybara-service Bot pushed a commit to tensorflow/tensorflow that referenced this pull request Sep 17, 2025
`com_google_protobuf` and `rules_cc` initialization was moved to `python_init_rules` macros.

The JAX repository update will be done [in the next step](jax-ml/jax#31709)

PiperOrigin-RevId: 803189916
copybara-service Bot pushed a commit to openxla/xla that referenced this pull request Sep 17, 2025
`com_google_protobuf` and `rules_cc` initialization was moved to `python_init_rules` macros.

The JAX repository update will be done [in the next step](jax-ml/jax#31709)

PiperOrigin-RevId: 803189916
copybara-service Bot pushed a commit to tensorflow/tensorflow that referenced this pull request Sep 17, 2025
`com_google_protobuf` and `rules_cc` initialization was moved to `python_init_rules` macros.

The JAX repository update will be done [in the next step](jax-ml/jax#31709)

PiperOrigin-RevId: 803189916
copybara-service Bot pushed a commit to openxla/xla that referenced this pull request Sep 17, 2025
`com_google_protobuf` and `rules_cc` initialization was moved to `python_init_rules` macros.

The JAX repository update will be done [in the next step](jax-ml/jax#31709)

PiperOrigin-RevId: 803189916
@copybara-service copybara-service Bot changed the title Update rules_python to v1.6 in JAX. Update rules_python to v1.6 in JAX. Sep 17, 2025
copybara-service Bot pushed a commit to openxla/xla that referenced this pull request Sep 17, 2025
`com_google_protobuf` and `rules_cc` initialization was moved to `python_init_rules` macros.

The JAX repository update will be done [in the next step](jax-ml/jax#31709)

PiperOrigin-RevId: 803189916
copybara-service Bot pushed a commit to openxla/xla that referenced this pull request Sep 18, 2025
`com_google_protobuf` and `rules_cc` initialization was moved to `python_init_rules` macros.

The JAX repository update will be done [in the next step](jax-ml/jax#31709)

PiperOrigin-RevId: 803189916
copybara-service Bot pushed a commit to tensorflow/tensorflow that referenced this pull request Sep 18, 2025
`com_google_protobuf` and `rules_cc` initialization was moved to `python_init_rules` macros.

The JAX repository update will be done [in the next step](jax-ml/jax#31709)

PiperOrigin-RevId: 803189916
@copybara-service copybara-service Bot force-pushed the test_805389759 branch 2 times, most recently from 17cf1ed to ecee7b1 Compare September 18, 2025 15:43
copybara-service Bot pushed a commit to openxla/xla that referenced this pull request Sep 18, 2025
`com_google_protobuf` and `rules_cc` initialization was moved to `python_init_rules` macros.

The JAX repository update will be done [in the next step](jax-ml/jax#31709)

PiperOrigin-RevId: 803189916
copybara-service Bot pushed a commit to tensorflow/tensorflow that referenced this pull request Sep 18, 2025
`com_google_protobuf` and `rules_cc` initialization was moved to `python_init_rules` macros.

The JAX repository update will be done [in the next step](jax-ml/jax#31709)

PiperOrigin-RevId: 803189916
copybara-service Bot pushed a commit to openxla/xla that referenced this pull request Sep 18, 2025
`com_google_protobuf` and `rules_cc` initialization was moved to `python_init_rules` macros.

The JAX repository update will be done [in the next step](jax-ml/jax#31709)

PiperOrigin-RevId: 808689003
copybara-service Bot pushed a commit to tensorflow/tensorflow that referenced this pull request Sep 18, 2025
`com_google_protobuf` and `rules_cc` initialization was moved to `python_init_rules` macros.

The JAX repository update will be done [in the next step](jax-ml/jax#31709)

PiperOrigin-RevId: 808689003
@copybara-service copybara-service Bot force-pushed the test_805389759 branch 3 times, most recently from de7a510 to 2cd57d2 Compare September 18, 2025 20:01
Also remove stale files from the wheel (see [link](1a91543#diff-674c34fb3618372dba0cf7f275754e7c4786c38beb8e7c7f8115e761265757d0)).

PiperOrigin-RevId: 808716151
@copybara-service copybara-service Bot merged commit 0624856 into main Sep 18, 2025
@copybara-service copybara-service Bot deleted the test_805389759 branch September 18, 2025 20:36
IgnotaYun pushed a commit to IgnotaYun/tensorflow that referenced this pull request Sep 22, 2025
`com_google_protobuf` and `rules_cc` initialization was moved to `python_init_rules` macros.

The JAX repository update will be done [in the next step](jax-ml/jax#31709)

PiperOrigin-RevId: 808689003
bhavya01 added a commit to pytorch/xla that referenced this pull request Oct 6, 2025
This PR updates the following pins:

- PyTorch:
pytorch/pytorch@928ac57
to
pytorch/pytorch@21fec65
(v2.9.0-rc5)
- OpenXLA:
openxla/xla@92f7b59
to
openxla/xla@9a9aa0e
- `libtpu`: 0.0.21 to 0.0.24
- JAX (and `jaxlib`): 0.7.1 to 0.8.0

**Key Changes:**

- `@python` was replaced by `@rules_python` at `BUILD` file (ref:
[jax-ml/jax#31709](jax-ml/jax#31709))
- `TF_ATTRIBUTE_NORETURN` was removed in favor of abseil (ref:
[openxla/xla#31699](openxla/xla#31699))
- Replaced include of `xla/pjrt/tfrt_cpu_pjrt_client.h` file by
`xla/pjrt/cpu/cpu_client.h` in `pjrt_registry.cpp`
([openxla/xla#30936](openxla/xla#30936))
- Moved the old `xla/tsl/platform/default/logging.*` to
`torch_xla/csrc/runtime/tsl_platform_logging.*`
- They were removed in
[openxla/xla#29477](openxla/xla#29477)
- Copied them here, temporarily. They should be removed once we update
our error throwing macros.
- Commented out a few macro definitions, avoiding macro re-definitions

**Update (Oct 3):**
- Add an OpenXLA patch for fixing `static_assert(false)` for GCC < 13
([ref](https://gcc.gnu.org/git/?p=gcc.git;a=commit;h=9944ca17c0766623bce260684edc614def7ea761))
- Removed the `flax` pin, since it does not overwrite `jax` anymore
- Removed `TPU*` prefix of `jax.experimental.pallas.tpu` components
(ref: [jax-ml/jax#29115](jax-ml/jax#29115))

---------

Co-authored-by: Bhavya Bahl <bbahl@google.com>
IgnotaYun pushed a commit to IgnotaYun/tensorflow that referenced this pull request Oct 15, 2025
`com_google_protobuf` and `rules_cc` initialization was moved to `python_init_rules` macros.

The JAX repository update will be done [in the next step](jax-ml/jax#31709)

PiperOrigin-RevId: 808689003
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.

1 participant