Skip to content

Error Handling: refactor the existing ComputationClient implementations to use status QOL functions. #9420

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Draft
wants to merge 3 commits into
base: master
Choose a base branch
from

Conversation

ysiraichi
Copy link
Collaborator

This PR modifies IfrtComputationClient and PjRtComputationClient in a similar way that XlaCoordinator was modified in #9386, by:

  • Creating a private struct PrivateUse for each of them, so as to make their constructor callable only in private contexts
  • Creating a Create() static member function that actually instantiates them, calling their constructor
  • Creating an Initialize() member function that is called by Create(), so as to initialize the class' members
  • Making these new member functions handle status types, checking and propagating them

@ysiraichi ysiraichi changed the title Refactor computation clients to use factory pattern Error Handling: refactor the existing ComputationClient implementations to use status QOL functions. Jun 28, 2025
@ysiraichi
Copy link
Collaborator Author

Blocked until #9419 is merged.

@ysiraichi ysiraichi force-pushed the ysiraichi/status-for-computation-clients branch from 328870d to 5ef2bb3 Compare June 28, 2025 14:41
@ysiraichi ysiraichi force-pushed the ysiraichi/status-for-pjrt-registry branch from 50d9c5a to e1bf39d Compare July 1, 2025 10:55
@ysiraichi ysiraichi force-pushed the ysiraichi/status-for-computation-clients branch from 5ef2bb3 to 4873c0d Compare July 1, 2025 10:55
@ysiraichi ysiraichi force-pushed the ysiraichi/status-for-pjrt-registry branch from e1bf39d to 2083d31 Compare July 1, 2025 15:09
@ysiraichi ysiraichi force-pushed the ysiraichi/status-for-computation-clients branch from 4873c0d to ee438d0 Compare July 1, 2025 15:10
@ysiraichi ysiraichi force-pushed the ysiraichi/status-for-pjrt-registry branch from 2083d31 to 40db714 Compare July 1, 2025 16:06
@ysiraichi ysiraichi force-pushed the ysiraichi/status-for-computation-clients branch from ee438d0 to fc8dcd5 Compare July 1, 2025 16:07
@ysiraichi ysiraichi force-pushed the ysiraichi/status-for-pjrt-registry branch from 40db714 to 9ebf41c Compare July 1, 2025 16:28
@ysiraichi ysiraichi force-pushed the ysiraichi/status-for-computation-clients branch from fc8dcd5 to 23263bf Compare July 1, 2025 16:28
@ysiraichi ysiraichi force-pushed the ysiraichi/status-for-pjrt-registry branch from 9ebf41c to 8499caf Compare July 1, 2025 18:14
@ysiraichi ysiraichi force-pushed the ysiraichi/status-for-computation-clients branch from 23263bf to 8ba721e Compare July 1, 2025 18:14
@ysiraichi ysiraichi force-pushed the ysiraichi/status-for-pjrt-registry branch from a4b9bd4 to c0e7506 Compare July 2, 2025 18:26
@ysiraichi ysiraichi force-pushed the ysiraichi/status-for-computation-clients branch from 8ba721e to df432d2 Compare July 2, 2025 18:27
@ysiraichi ysiraichi force-pushed the ysiraichi/status-for-pjrt-registry branch 3 times, most recently from c4595dc to 3c1289f Compare July 10, 2025 12:06
Update IFRT and PjRt computation clients to use `Create()` factory
methods:
- Replace constructors with factory methods that return `StatusOr<T>`
- Use `ConsumeAndMaybeThrow` for `XlaCoordinator::Create` integration
- Improved error handling with proper status propagation
@ysiraichi ysiraichi force-pushed the ysiraichi/status-for-computation-clients branch from df432d2 to 9c19b16 Compare July 10, 2025 19:05
@ysiraichi ysiraichi changed the base branch from ysiraichi/status-for-pjrt-registry to master July 10, 2025 19:05
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