fix comment

Author: qyh111

examples/ucm_config_example.yaml

@@ -14,10 +14,6 @@ ucm_connectors:
       storage_backends: "/mnt/test"
       transferIoDirect: false
 
-  - ucm_connector_name: "UcmLocalStore"
-    ucm_connector_config:
-      cache_size: 12288
-
 load_only_first_rank: false
 
 # Sparse attention configuration

ucm/integration/vllm/ucm_connector.py

@@ -58,15 +58,14 @@ class RequestHasher:
 
     def __init__(self):
         if RequestHasher._SEED_HASH is None:
-            RequestHasher._SEED_HASH = self.md5("UCM_HASH_SEED")
+            RequestHasher._SEED_HASH = self("UCM_HASH_SEED")
 
-    @staticmethod
-    def md5(input_data) -> int:
+    def __call__(self, input_data) -> int:
         input_bytes = pickle.dumps(input_data, protocol=pickle.HIGHEST_PROTOCOL)
         md5_bytes = hashlib.md5(input_bytes).digest()
         return int.from_bytes(md5_bytes, byteorder="big")
 
-    def __call__(self, block_size: int, request: "Request") -> list[str]:
+    def process(self, block_size: int, request: "Request") -> list[str]:
         token_ids = request.all_token_ids
 
         ret = []
@@ -82,7 +81,7 @@ def __call__(self, block_size: int, request: "Request") -> list[str]:
                 parent_block_hash_value = RequestHasher._SEED_HASH
 
             block_token_ids_tuple = tuple(block_token_ids)
-            hash_value = self.md5((parent_block_hash_value, block_token_ids_tuple))
+            hash_value = self((parent_block_hash_value, block_token_ids_tuple))
             parent_block_hash_value = hash_value
             ret.append(str(hash_value))
 
@@ -182,7 +181,7 @@ def get_num_new_matched_tokens(
         assert num_computed_tokens % self.block_size == 0
         hbm_hit_block_num = num_computed_tokens // self.block_size
 
-        ucm_block_ids = self.request_hasher(self.block_size, request)
+        ucm_block_ids = self.request_hasher.process(self.block_size, request)
 
         external_block_ids = ucm_block_ids[hbm_hit_block_num:]
         if not external_block_ids:
@@ -449,7 +448,7 @@ def start_load_kv(self, forward_context: "ForwardContext", **kwargs) -> None:
             ucm_block_ids, vllm_block_ids = request.load_block_ids
             if self.rank != 0:
                 for i, ucm_block_id in enumerate(ucm_block_ids):
-                    ucm_block_ids[i] = str(RequestHasher.md5((ucm_block_id, self.rank)))
+                    ucm_block_ids[i] = str(self.request_hasher((ucm_block_id, self.rank)))
             ucm_total_block_ids, ucm_offsets, dst_tensor_addr = self._generate_task(
                 vllm_block_ids, ucm_block_ids
             )
@@ -498,7 +497,7 @@ def wait_for_save(self) -> None:
             ucm_block_ids, vllm_block_ids = request.dump_block_ids
             if self.rank != 0:
                 for i, ucm_block_id in enumerate(ucm_block_ids):
-                    ucm_block_ids[i] = str(RequestHasher.md5((ucm_block_id, self.rank)))
+                    ucm_block_ids[i] = str(self.request_hasher((ucm_block_id, self.rank)))
             rets = self.store.create(ucm_block_ids)
             end = 0
             for i, ret in enumerate(rets):