jax
CI: 01/09/25 upstream sync
#197
Merged

CI: 01/09/25 upstream sync #197

charleshofer merged 15 commits into rocm-main from ci-upstream-sync-83_1
github-actions
vfdev-5 Avoid adding conflicting --repo_env=HERMETIC_PYTHON_VERSION= to bazel…
70e06c2d
sharadmv [Pallas] Add empty/empty_like helper functions
c1a60c67
yashk2810 [sharding_in_types] Functions like einsum, reshape, broadcast_in_dim,…
3848f0d2
justinjfu [Mosaic GPU] Allow multiple indexing on refs
d99a637d
Google-ML-Automation Merge pull request #25527 from vfdev-5:single-python-version-build-py
6e1f060a
yashk2810 Match the behavior on single host wrt multi-host if tiled=False. Fixe…
b3833dc7
bixia1 Implement the extension to the custom_partitioning API.
c4ac0dd6
vfdev-5 Added 3.13 ft requirements lock file and updated WORKSPACE
00806dda
hawkinsp Make api_test.py work when test cases are run using multiple threads.
e20523c2
Google-ML-Automation Merge pull request #25786 from vfdev-5:add-313-ft-configs
196eec82
bartchr808 #sdy add repr for Sdy ArraySharding and DimSharding
cbcc883e
yashk2810 Respect the original memory kind on reshape, transpose and replicate …
fb832afc
yashk2810 Make sharding_in_types work with Shardy
b2b38679
bazel visibility change
640cb009
github-actions[bot] github-actions[bot] enabled auto-merge 356 days ago
charleshofer Commit to trigger CI
9d34a49d
disabled auto-merge 356 days ago
Manually disabled by user
charleshofer
charleshofer approved these changes on 2025-01-09
charleshofer charleshofer merged c22bed55 into rocm-main 356 days ago
charleshofer charleshofer deleted the ci-upstream-sync-83_1 branch 356 days ago

Login to write a write a comment.

Login via GitHub

Reviewers
Assignees
No one assigned
Labels
Milestone