Skip to content

Improve multi-NIC discovery and NUMA resolution - #923

Merged
copybara-service[bot] merged 1 commit into
mainfrom
test_981036297
Sep 17, 2026
Merged

copybara-service[bot] merged 1 commit into
mainfrom
test_981036297

Conversation

@copybara-service

@copybara-service copybara-service Bot commented Sep 14, 2026

Copy link
Copy Markdown

Improve multi-NIC discovery and NUMA resolution

Expose TPU_RAIDEN_DATA_NICS in case the NIC auto-discovery doesn't work for user.

@copybara-service copybara-service Bot changed the title Support multi-NIC discovery, multi-process NUMA resolution via LOCAL_RANK, and plumb bind_ip Improve multi-NIC discovery and NUMA resolution Sep 14, 2026
- Expose `TPU_RAIDEN_DATA_NICS` allowlist in case NIC auto-discovery doesn't work.
- Make source IP binding opt-in via `TPU_RAIDEN_ENABLE_SOURCE_IP_BIND`.
- Resolve TPU NUMA nodes natively from PJRT device attributes.

PiperOrigin-RevId: 982922200
@copybara-service
copybara-service Bot merged commit a4db5e7 into main Sep 17, 2026
1 check passed
@copybara-service
copybara-service Bot deleted the test_981036297 branch September 17, 2026 04:40
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