Skip to content

Conversation

@zpcore
Copy link
Member

@zpcore zpcore commented Apr 16, 2025

xb.call_jax now hashing with jax._src.config.

@zpcore zpcore marked this pull request as ready for review April 16, 2025 00:05
@zpcore zpcore requested review from qihqi and tengyifei April 16, 2025 00:05
@tengyifei
Copy link
Collaborator

Just a heads up: I submitted @assume_pure in https://github.com/pytorch/xla/pull/8962/files#diff-088930d7c2ec653486390434bafbb93fed86eb38c39c14a997f116276976ba9eR154 and you might need to rebase then rename my few usages of the cache_miss function too!

@tengyifei
Copy link
Collaborator

Unfortunately you'd have to rebase again because our CI broke today and just got fixed by #8979

@zpcore zpcore enabled auto-merge (squash) April 16, 2025 16:37
@zpcore zpcore merged commit a48d6a9 into master Apr 16, 2025
23 of 24 checks passed
@zpcore zpcore deleted the piz/cj-hashing branch April 16, 2025 19:42
@zpcore zpcore mentioned this pull request Apr 16, 2025
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.

3 participants