[HiCache][HybridModel]: Support mamba state offloading & HybridCacheController (#20457)

Co-authored-by: pansicheng <sicheng.pan.chn@gmail.com>
Co-authored-by: 晟海 <huangtingwei.htw@antgroup.com>
Co-authored-by: ispobock <ispobaoke@gmail.com>
This commit is contained in:
hzh0425
2026-03-23 20:02:50 -07:00
committed by GitHub
co-authored by pansicheng 晟海 ispobock
parent 2b1d3c935e
commit 0986bed8e2
13 changed files with 2012 additions and 220 deletions
+11
View File
@@ -529,6 +529,8 @@ class WorkloadGenerator:
f"({expected} clients), releasing {len(next_round_reqs)} "
f"requests for round {current_barrier_round + 1}"
)
self._send_heartbeat(input_len=100, output_len=100)
time.sleep(10)
for req in next_round_reqs:
self.ready_queue.append(req)
next_round_reqs = []
@@ -541,6 +543,15 @@ class WorkloadGenerator:
print(f"Error processing response for client {client_id}: {e}")
continue
def _send_heartbeat(self, input_len=100, output_len=20):
"""Send a small heartbeat request to the server."""
heartbeat_input = [1] * input_len
payload = gen_payload(heartbeat_input, output_len, self.lora_path)
try:
requests.post(self.url, json=payload, timeout=30)
except Exception as e:
print(f"Heartbeat request failed: {e}")
def run(self):
request_thread = threading.Thread(target=self.request_sender, daemon=True)
response_thread = threading.Thread(target=self.response_handler, daemon=True)