Skip to content

Comments

[xla:pjrt] Unify AsyncWorkRunner API with tsl::Executor#38135

Open
ezhulenev wants to merge 1 commit intoopenxla:mainfrom
ezhulenev:async-work-runner-0
Open

[xla:pjrt] Unify AsyncWorkRunner API with tsl::Executor#38135
ezhulenev wants to merge 1 commit intoopenxla:mainfrom
ezhulenev:async-work-runner-0

Conversation

@ezhulenev
Copy link
Contributor

@ezhulenev ezhulenev commented Feb 21, 2026

We already have tsl::Executor that solves the same problem in many other APIs, migrate PJRT to standard API with a path for further cleanup.

  1. AsyncWorkRunner is a tsl::Executor
  2. Implement ExecuteWhenReady once because all implementations use the same RunWhenReady

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