Skip to content

Commit c426243

Browse files
committed
Testing
1 parent 9d2c513 commit c426243

File tree

2 files changed

+13
-4
lines changed

2 files changed

+13
-4
lines changed

src/python_be.cc

Lines changed: 8 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1303,6 +1303,14 @@ ModelInstanceState::ResponseSendDecoupled(
13031303
SetErrorForResponseSendMessage(
13041304
send_message_payload, WrapTritonErrorInSharedPtr(error), error_message);
13051305
}
1306+
1307+
if (send_message_payload->flags == TRITONSERVER_RESPONSE_COMPLETE_FINAL) {
1308+
// Delete response factory
1309+
std::unique_ptr<
1310+
TRITONBACKEND_ResponseFactory, backend::ResponseFactoryDeleter>
1311+
response_factory(reinterpret_cast<TRITONBACKEND_ResponseFactory*>(
1312+
send_message_payload->response_factory_address));
1313+
}
13061314
}
13071315

13081316
TRITONSERVER_Error*

src/response_sender.cc

Lines changed: 5 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -47,10 +47,11 @@ ResponseSender::ResponseSender(
4747

4848
ResponseSender::~ResponseSender()
4949
{
50-
std::unique_ptr<Stub>& stub = Stub::GetOrCreateInstance();
51-
stub->EnqueueCleanupId(
52-
reinterpret_cast<void*>(response_factory_address_),
53-
PYTHONSTUB_DecoupledResponseFactoryCleanup);
50+
// std::cerr << "===== ResponseSender::~ResponseSender() =====" << std::endl;
51+
// std::unique_ptr<Stub>& stub = Stub::GetOrCreateInstance();
52+
// stub->EnqueueCleanupId(
53+
// reinterpret_cast<void*>(response_factory_address_),
54+
// PYTHONSTUB_DecoupledResponseFactoryCleanup);
5455
}
5556

5657
void

0 commit comments

Comments
 (0)