Skip to content

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

Open
prishajain1 wants to merge 1 commit into
mainfrom
prisha/fix-tpu-device-discovery
Open

Fix: bump libtpu version to resolve TPU v7x device initialization failure#471
prishajain1 wants to merge 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.

- 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
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.

1 participant