|
14 | 14 | class HiRadixCache(RadixCache):
|
15 | 15 | def __init__(self, unique_name, total_token_num, rank_in_node, mem_manager, max_seq_length):
|
16 | 16 | super().__init__(unique_name, total_token_num, rank_in_node, mem_manager)
|
17 |
| - logger.info(f"Initializing HiRadixCache") |
| 17 | + logger.info("Initializing HiRadixCache") |
18 | 18 | self.rank_in_node = rank_in_node
|
19 | 19 | try:
|
20 | 20 | self.is_hi_radix_cache = True
|
@@ -66,7 +66,7 @@ def wait_till_finish(self, req_id):
|
66 | 66 | # with self.moving_lock:
|
67 | 67 | # if (not self.start_store_task) and self.write_task is not None:
|
68 | 68 | # if self.write_task.ready():
|
69 |
| - # logger.info(f"HiCache of [{self.rank_in_node}]: stored one kvcache with len = {self.hi_cache_buffer_len}") |
| 69 | + # logger.info(f"HiCache of [{self.rank_in_node}]: stored len = {self.hi_cache_buffer_len}") |
70 | 70 | # self.start_store_task = True # ensure ready => start new only one kvcache stores
|
71 | 71 | # do_copy = True
|
72 | 72 | # elif self.write_task is None and self.starting:
|
@@ -119,7 +119,7 @@ def match_prefix(self, key, update_refs=False):
|
119 | 119 | buffers = self.mem_manager.alloc(max_len)
|
120 | 120 | before_pull_time = time.time()
|
121 | 121 | logger.info(
|
122 |
| - f"HiCache of [{self.rank_in_node}]: No.2.5 Before disk pull took {before_pull_time - hi_cache_query_time}" |
| 122 | + f"HiCache of [{self.rank_in_node}]: No.2.5 Before pull took {before_pull_time - hi_cache_query_time}" |
123 | 123 | )
|
124 | 124 | read_task = self.py_cache_service.create(tokens=key[:max_len], kv_page_indexer=buffers, mode="r")
|
125 | 125 | while not read_task.ready():
|
|
0 commit comments