Skip to content

[pull] master from tensorflow:master#1698

Merged
pull[bot] merged 11 commits into
GesuBackups:masterfrom
tensorflow:master
Apr 6, 2026
Merged

[pull] master from tensorflow:master#1698
pull[bot] merged 11 commits into
GesuBackups:masterfrom
tensorflow:master

Conversation

@pull

@pull pull Bot commented Apr 6, 2026

Copy link
Copy Markdown

See Commits and Changes for more details.


Created by pull[bot] (v2.0.0-alpha.4)

Can you help keep this open source service alive? 💖 Please sponsor : )

dependabot Bot and others added 11 commits March 30, 2026 18:46
Bumps [pygments](https://github.com/pygments/pygments) from 2.19.1 to 2.20.0.
- [Release notes](https://github.com/pygments/pygments/releases)
- [Changelog](https://github.com/pygments/pygments/blob/master/CHANGES)
- [Commits](pygments/pygments@2.19.1...2.20.0)

---
updated-dependencies:
- dependency-name: pygments
  dependency-version: 2.20.0
  dependency-type: direct:production
...

Signed-off-by: dependabot[bot] <support@github.com>
Bumps [pygments](https://github.com/pygments/pygments) from 2.19.1 to 2.20.0.
- [Release notes](https://github.com/pygments/pygments/releases)
- [Changelog](https://github.com/pygments/pygments/blob/master/CHANGES)
- [Commits](pygments/pygments@2.19.1...2.20.0)

---
updated-dependencies:
- dependency-name: pygments
  dependency-version: 2.20.0
  dependency-type: direct:production
...

Signed-off-by: dependabot[bot] <support@github.com>
This flag preset will continue to be developed with fast compilation times and numerical stability in mind as the top goals (runtime performance only a secondary goal). Expect tradeoffs similar to XX% compilation time for X% runtime to occur under this flag.

Currently, it just sets LLVM codegen opt to O1, disables platform dependent math, and sets `flatten_after_fusion` to true.

PiperOrigin-RevId: 895343655
…/requirements_updater/numpy1_requirements/pygments-2.20.0

PiperOrigin-RevId: 895376530
PiperOrigin-RevId: 895385907
PiperOrigin-RevId: 895403161
… platform.

The simplest option is to look up the default compiler and then ask if it's a
PhaseCompiler by downcasting.

PiperOrigin-RevId: 895410872
Imported from GitHub PR openxla/xla#40418

Replace the `bool is_p2p` flag with a strongly-typed `CommunicationId` integer across `GpuCliqueKey`, `CollectiveThunk`, and all related APIs.

This lays the groundwork for a more flexible communication model where multiple different communication kinds can overlap on the same set of devices. The actual overlapping support is coming in a follow-up change. On the runtime side we already have `ExecutionStreamId` and `CommunicationStreamId`, this is the corresponding part of collectives API.

For now hardcode p2p communication as `CommunicationId(1)`.

## What changed

- **`GpuCliqueKey`**: `bool is_p2p` → `CommunicationId communication_id`.
- **`CollectiveThunk`**: `bool is_p2p_` / `IsP2PCollective()` → `CommunicationId communication_id_` / `communication_id()`.
- **`GetGpuCliqueKey` / `GetCollectiveGpuCliqueKey`**: `bool is_p2p` param → `CommunicationId`.
- **All subclass thunks**: P2P thunks pass `CommunicationId(1)`, collectives use default `CommunicationId(0)`.
- **`gpu_clique_key.h`**: Removed `IsP2PStreamKind`, `GetCollectiveStreamId`, `CollectiveStreamId`, `kAsyncStreamTotal`.
- **`gpu_executable.cc`**: Localized `kAsyncStreamTotal`, minor include/rename cleanup.
Copybara import of the project:

--
d8570ac819761309d1661665e8ccac4f95c34a2b by Eugene Zhulenev <ezhulenev@openxla.org>:

[xla:gpu] Add CommunicationId to identify communication types

Merging this change closes #40418

PiperOrigin-RevId: 895442107
@pull pull Bot locked and limited conversation to collaborators Apr 6, 2026
@pull pull Bot added the ⤵️ pull label Apr 6, 2026
@pull pull Bot merged commit 032075b into GesuBackups:master Apr 6, 2026
Sign up for free to subscribe to this conversation on GitHub. Already have an account? Sign in.

Projects

None yet

Development

Successfully merging this pull request may close these issues.

7 participants