Skip to content

Fix: bump libtpu version to resolve TPU v7x device initialization failure - #471

Merged
mbohlool merged 1 commit into
mainfrom
prisha/fix-tpu-device-discovery
Sep 11, 2026
Merged

Fix: bump libtpu version to resolve TPU v7x device initialization failure#471
mbohlool merged 1 commit into
mainfrom
prisha/fix-tpu-device-discovery

Conversation

@prishajain1

Copy link
Copy Markdown
Collaborator

Problem

Nightly regression tests running on Ironwood (tpu7x) using maxdiffusion_jax_nightly:latest fail on startup during jax.device_count() with:

WARNING: No hardware is found. Using default TPU version: ba16c7433
RuntimeError: Unable to initialize backend 'tpu': UNKNOWN: TPU initialization failed: No ba16c7433 device found.

Fix:

  • Bump minimum libtpu to >=0.0.42.1 in requirements and setup.sh to ensure compatibility with TPU v7x (Ironwood) hardware.
  • Avoid conflicting libtpu_nightly installations in setup.sh that collide with libtpu.

@prishajain1
prishajain1 requested a review from entrpn as a code owner September 3, 2026 04:42
@gemini-code-assist

Copy link
Copy Markdown

Warning

You have reached your daily quota limit. Please wait up to 24 hours and I will start processing your requests again!

@github-actions

github-actions Bot commented Sep 3, 2026

Copy link
Copy Markdown

@prishajain1
prishajain1 requested a review from mbohlool September 3, 2026 04:47
- Bump minimum libtpu to >=0.0.42.1 in requirements and setup.sh to ensure compatibility with TPU v7x (Ironwood) hardware.
- Avoid conflicting libtpu_nightly installations in setup.sh that collide with libtpu.
@prishajain1
prishajain1 force-pushed the prisha/fix-tpu-device-discovery branch from 7ddaadb to fd01a47 Compare September 10, 2026 17:10
@mbohlool
mbohlool merged commit a4ff3ae into main Sep 11, 2026
15 checks passed
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Projects

None yet

Development

Successfully merging this pull request may close these issues.

3 participants