xla
Update PyTorch and XLA pin.
#9668
Merged
Overview
Commits
15
Changes
View On GitHub

Update PyTorch and XLA pin. #9668

bhavya01 merged 15 commits into master from ysiraichi/update-openxla-pin
ysiraichi
ysiraichi Update pins.
9c30add1
ysiraichi Add commit reference.
3042439d
ysiraichi Fix lint.
8e91c058
ysiraichi Add OpenXLA patch.
8af438ef
ysiraichi Unpin flax, but check for conflicts.
6ea29021
ysiraichi Remove TPU prefix.
488d8c66
ysiraichi Remove pip check.
1b508b96
ysiraichi Fix sharding test HLO.
11fa7bec
ysiraichi Fix lint.
15d8817b
ysiraichi Fix test.
66164c8a
bhavya01 fix xla sharding tests
8a5a0d53
bhavya01 Update opt-barrier test
2f1e61ce
bhavya01 Remove flax install from CI
bf1e0e78
bhavya01 Remove shard_as test
63a46004
qihqi
qihqi commented on 2025-10-06
bhavya01 Add back flax
e7f0765d
bhavya01
bhavya01 approved these changes on 2025-10-06
bhavya01 bhavya01 enabled auto-merge (squash) 97 days ago
bhavya01 bhavya01 merged 11590c1c into master 97 days ago

Login to write a write a comment.

Login via GitHub

Reviewers
Assignees
No one assigned
Labels
None yet
Milestone
No milestone
Terms Of Use Privacy Policy Refund Policy Impressum