[Perf]kv_block_size as well as transferIoSize are calculated rather than configured (#196)

* kv_block_size as well as transferIoSize are calculated rather than configured

---------

Co-authored-by: root <fenghao78@huawei.com>

Author: UESTC-AHao

ucm/integration/vllm/uc_connector.py

@@ -109,6 +109,10 @@ def __init__(self, vllm_config: "VllmConfig", role: KVConnectorRole):
         self._need_load_reqs: dict[str, Union[list[int], list[Task]]] = {}
         self._load_failed_reqs: set[str] = set()
         self._load_req_to_blocks: dict[str, set[int]] = {}
+        self.num_head = vllm_config.model_config.get_num_kv_heads(
+            vllm_config.parallel_config
+        )
+        self.head_size = vllm_config.model_config.get_head_size()
         if role == KVConnectorRole.WORKER:
             self._initialize_dataoffset(vllm_config)
         if (
@@ -131,6 +135,20 @@ def __init__(self, vllm_config: "VllmConfig", role: KVConnectorRole):
             config["role"] = (
                 "scheduler" if role == KVConnectorRole.SCHEDULER else "worker"
             )
+            config_base = self.block_size * self.element_size * self.head_size
+            config["kv_block_size"] = (
+                config_base
+                * self.num_layers
+                * (1 if self.is_mla else self.num_head * self.total_tp_size * 2)
+            )
+            config["transferIoSize"] = config_base * (
+                1 if self.is_mla else self.num_head
+            )
+            logger.info(
+                "kv_block_size = %d, transferIoSize = %d,",
+                config["kv_block_size"],
+                config["transferIoSize"],
+            )
             logger.info("init UCConnectorImpl, connector: %s", name)
             self.connector = UcmConnectorFactory.create_connector(name, config)
         else: