Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Update libtpu pin to 2024/12/05 / 0.0.5 stable #8461

Merged
merged 1 commit into from
Dec 6, 2024
Merged

Update libtpu pin to 2024/12/05 / 0.0.5 stable #8461

merged 1 commit into from
Dec 6, 2024

Conversation

tengyifei
Copy link
Collaborator

@tengyifei tengyifei commented Dec 6, 2024

The stable libtpu is qualified with more tests compared to nightly tpu. In particular, it has also been qualified with all JAX unit tests, which is beneficial since we depend on JAX.

This PR switches to stable libtpu at HEAD. Later I'll cut a 2.6 branch and that will also get the stable libtpu.

Depending on a stable libtpu means we don't need as much churn of updating the pin. If we later find a urgent need to adopt a specific libtpu CL, we could still switch back to a nightly at HEAD without issues.

@tengyifei tengyifei added the tpuci label Dec 6, 2024
@tengyifei tengyifei force-pushed the test branch 3 times, most recently from 4fc0e0f to cd83871 Compare December 6, 2024 01:44
@tengyifei tengyifei changed the title test Update libtpu pin to 2024/12/05 / 0.0.5 stable Dec 6, 2024
@tengyifei tengyifei requested review from qihqi and lsy323 December 6, 2024 04:19
@tengyifei tengyifei marked this pull request as ready for review December 6, 2024 04:22
@tengyifei tengyifei enabled auto-merge (squash) December 6, 2024 18:59
@@ -64,10 +64,11 @@

base_dir = os.path.dirname(os.path.abspath(__file__))

_date = '20241122'
Copy link
Collaborator

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

nits: Maybe adding a flag for using the stable version, and add the .dev{_date} conditionally. This might make future pin update using nightly easier.

Copy link
Collaborator Author

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

oops, github auto-merge got too happy. i'll upload a fixup PR

@tengyifei tengyifei merged commit 85cf18a into master Dec 6, 2024
12 checks passed
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
Projects
None yet
Development

Successfully merging this pull request may close these issues.

2 participants