[None][fix] Fix ready signal in NIXL backend (#10000)

chuangz0 · web-flow · commit e0b2a9430983 · 2025-12-19T09:43:40.000+08:00
Signed-off-by: Chuang Zhu &lt;111838961+chuangz0@users.noreply.github.com&gt;
diff --git a/cpp/tensorrt_llm/executor/cache_transmission/agent_utils/connection.cpp b/cpp/tensorrt_llm/executor/cache_transmission/agent_utils/connection.cpp
@@ -231,7 +231,7 @@ bool AgentConnection::recvReadySignal(DataContext const& ctx) const
 {
     ReadySignalInfo readySignalInfo{mAgentName, ctx, false};
     mAgentConnectionManager->waitForReadySignal(mRemoteAgentName, readySignalInfo);
-    return true;
+    return readySignalInfo.mIsReady;
 }
 
 AgentConnectionManager::AgentConnectionManager(
diff --git a/tests/integration/test_lists/test-db/l0_sanity_check.yml b/tests/integration/test_lists/test-db/l0_sanity_check.yml
@@ -34,3 +34,5 @@ l0_sanity_check:
       - examples/test_llm_api_with_mpi.py::test_llm_api_single_gpu_with_mpirun[TinyLlama-1.1B-Chat-v1.0] ISOLATION
       - unittest/others/test_kv_cache_transceiver.py::test_kv_cache_transceiver_single_process[NIXL-mha-ctx_fp16_gen_fp16]
       - unittest/others/test_kv_cache_transceiver.py::test_kv_cache_transceiver_single_process[UCX-mha-ctx_fp16_gen_fp16]
+      - unittest/others/test_kv_cache_transceiver.py::test_cancel_request_in_transmission[mha]
+      - unittest/others/test_kv_cache_transceiver.py::test_cancel_request_in_transmission[mla]

Original file line number	Diff line number	Diff line change
`@@ -231,7 +231,7 @@ bool AgentConnection::recvReadySignal(DataContext const& ctx) const`
`231`	`231`	`{`
`232`	`232`	`ReadySignalInfo readySignalInfo{mAgentName, ctx, false};`
`233`	`233`	`mAgentConnectionManager->waitForReadySignal(mRemoteAgentName, readySignalInfo);`
`234`		`- return true;`
	`234`	`+ return readySignalInfo.mIsReady;`
`235`	`235`	`}`
`236`	`236`
`237`	`237`	`AgentConnectionManager::AgentConnectionManager(`