Skip to content

Commit

Permalink
Use a single SEND_END point
Browse files Browse the repository at this point in the history
  • Loading branch information
tgerdesnv committed Mar 5, 2024
1 parent 0e06c11 commit fa0fcb7
Showing 1 changed file with 2 additions and 6 deletions.
8 changes: 2 additions & 6 deletions src/c++/perf_analyzer/client_backend/openai/openai_client.cc
Original file line number Diff line number Diff line change
Expand Up @@ -84,10 +84,8 @@ ChatCompletionClient::RequestProvider(
request->GetNextInput(
reinterpret_cast<uint8_t*>(contents), size * nmemb, &input_bytes);

if (input_bytes == 0) {
request->timer_.CaptureTimestamp(
triton::client::RequestTimers::Kind::SEND_END);
}
request->timer_.CaptureTimestamp(
triton::client::RequestTimers::Kind::SEND_END);

return input_bytes;
}
Expand Down Expand Up @@ -185,8 +183,6 @@ ChatCompletionClient::AsyncInfer(
raw_request->timer_.CaptureTimestamp(
triton::client::RequestTimers::Kind::SEND_START);
Send(multi_easy_handle, std::move(request));
raw_request->timer_.CaptureTimestamp(
triton::client::RequestTimers::Kind::SEND_END);
return Error::Success;
}

Expand Down

0 comments on commit fa0fcb7

Please sign in to comment.