Skip to content

Conversation

kakwok
Copy link

@kakwok kakwok commented Apr 2, 2025

PR description:

Implements a RetryActionBase class for different types of action to be taken upon the need to retry.
SonicClientBase will have a list of RetryAction pointers initialized at construction.
The client will retry the list of actions one-by-one until every actions is no-longer suitable to retry.

An example configuration may look like this:

   Client = cms.PSet(
        mode = cms.string("Sync"),
        factor = cms.int32(-1),
        wait = cms.int32(10),
        allowedTries = cms.untracked.uint32(0),
        fails = cms.uint32(0),
        Retry = cms.VPSet(
              Action1 = cms.PSet(
                  retryType = cms.string("RetrySameServerAction"),
                  allowedTries = cms.untracked.uint32(4)
              ),
                Action2 = cms.PSet(
                  retryType = cms.string("RetrySomeOthersAction"),
                    ...
              ),
        )
    ),

@@ -65,6 +65,7 @@ class TritonClient : public SonicClient<TritonInputMap, TritonOutputMap> {
bool handle_exception(F&& call);

void reportServerSideStats(const ServerSideStats& stats) const;
void updateServer(std::string serverName);
Copy link

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

const std::string&

@@ -574,6 +561,26 @@ inference::ModelStatistics TritonClient::getServerSideStatus() const {
return inference::ModelStatistics{};
}

void TritonClient::updateServer(std::string serverName){
Copy link

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

const std::string&

useSharedMemory = cms.untracked.bool(not options.noShm),
compression = cms.untracked.string(options.compression),
Retry = cms.VPSet(
Copy link

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

should rebase on latest 15_1_X prerelease (to incorporate cms-sw#47748)

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.

2 participants