Skip to content

Commit

Permalink
Improve decoupled shm handling (#337)
Browse files Browse the repository at this point in the history
* [DO NOT MERGE] Add shm trace util

* [DO NOT MERGE] Expand shm leak util naming to ipc load

* Revert "[DO NOT MERGE] Expand shm leak util naming to ipc load"

This reverts commit 68906f2.

* Revert "[DO NOT MERGE] Add shm trace util"

This reverts commit 37824ce.

* Fix decoupled shared memory leak
  • Loading branch information
kthui authored Jan 19, 2024
1 parent 9d67dc3 commit 37d2902
Showing 1 changed file with 1 addition and 0 deletions.
1 change: 1 addition & 0 deletions src/python_be.cc
Original file line number Diff line number Diff line change
Expand Up @@ -1328,6 +1328,7 @@ ModelInstanceState::ProcessRequestsDecoupled(

AllocatedSharedMemory<ResponseBatch> response_batch =
Stub()->ShmPool()->Load<ResponseBatch>(received_message_->Args());
received_message_.reset();

uint64_t compute_end_ns = 0;
SET_TIMESTAMP(compute_end_ns);
Expand Down

0 comments on commit 37d2902

Please sign in to comment.