Skip to content

Commit

Permalink
Add inference request cancellation APIs (#249)
Browse files Browse the repository at this point in the history
* Fix state transitions for re-running requests (#251)

* Add backend/server APIs

* Implement the cancellation APIs

* Only store the state in response factory

* Add unit testing for request cancellation

* Add test

* Add cancellation status

* Add testing for cancelling a request after release

* Handle request re-use

* Enable request reuse test

* Add staged changes

* Add temporary fix for the request state bug

---------

Co-authored-by: Ryan McCormick <rmccormick@nvidia.com>
  • Loading branch information
Tabrizian and rmccorm4 committed Sep 13, 2023
1 parent 2e2a617 commit b34c261
Showing 1 changed file with 8 additions and 0 deletions.
8 changes: 8 additions & 0 deletions src/test/CMakeLists.txt
Original file line number Diff line number Diff line change
Expand Up @@ -592,6 +592,14 @@ set_target_properties(
INSTALL_RPATH ""
)

target_include_directories(
request_cancellation_test
PRIVATE
${CMAKE_CURRENT_SOURCE_DIR}/..
${CMAKE_CURRENT_SOURCE_DIR}/../../include
${GTEST_INCLUDE_DIRS}
)

target_link_libraries(
request_cancellation_test
PRIVATE
Expand Down

0 comments on commit b34c261

Please sign in to comment.