fix broadcast

Author: qyh111

ucm/integration/vllm/ucm_connector.py

@@ -477,7 +477,7 @@ def start_load_kv(self, forward_context: "ForwardContext", **kwargs) -> None:
                 )
             else:
                 request_to_task[request_id] = None
-            req_broadcast_addr[request_id] = tensors
+            req_broadcast_addr[request_id] = [t for row in tensors for t in row] 
 
         for request_id, task in request_to_task.items():
             # TODO error handling

ucm/store/test/e2e/pcstore_embed.py

@@ -106,7 +106,7 @@ def store_all_hashes(hashes):
 
 
 def main():
-    storage_backends = "/home/share/qyh-test"
+    storage_backends = "."
     block_number = 4096
     device_id = 1
     block_dim = 576

ucm/store/test/e2e/pcstore_fetch.py

@@ -83,7 +83,7 @@ def fetch(
 
 
 def main():
-    storage_backends = "/home/share/qyh-test"
+    storage_backends = "."
     device_id = 1
     block_dim = 576
     block_len = 64