|
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