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