Unverified Commit 0b5b5d76 authored by AlpinDale's avatar AlpinDale Committed by GitHub
Browse files

[Frontend] Log the maximum supported concurrency (#8831)

parent cdc72e3c
...@@ -56,6 +56,10 @@ class DistributedGPUExecutor(GPUExecutor): ...@@ -56,6 +56,10 @@ class DistributedGPUExecutor(GPUExecutor):
# have GPUs. # have GPUs.
logger.info("# GPU blocks: %d, # CPU blocks: %d", num_gpu_blocks, logger.info("# GPU blocks: %d, # CPU blocks: %d", num_gpu_blocks,
num_cpu_blocks) num_cpu_blocks)
max_concurrency = (num_gpu_blocks * self.cache_config.block_size /
self.model_config.max_model_len)
logger.info("Maximum concurrency for %s tokens per request: %.2fx",
self.model_config.max_model_len, max_concurrency)
self.cache_config.num_gpu_blocks = num_gpu_blocks self.cache_config.num_gpu_blocks = num_gpu_blocks
self.cache_config.num_cpu_blocks = num_cpu_blocks self.cache_config.num_cpu_blocks = num_cpu_blocks
......
...@@ -121,6 +121,10 @@ class GPUExecutor(ExecutorBase): ...@@ -121,6 +121,10 @@ class GPUExecutor(ExecutorBase):
# remains to abstract away the device for non-GPU configurations. # remains to abstract away the device for non-GPU configurations.
logger.info("# GPU blocks: %d, # CPU blocks: %d", num_gpu_blocks, logger.info("# GPU blocks: %d, # CPU blocks: %d", num_gpu_blocks,
num_cpu_blocks) num_cpu_blocks)
max_concurrency = (num_gpu_blocks * self.cache_config.block_size /
self.model_config.max_model_len)
logger.info("Maximum concurrency for %s tokens per request: %.2fx",
self.model_config.max_model_len, max_concurrency)
self.driver_worker.initialize_cache(num_gpu_blocks, num_cpu_blocks) self.driver_worker.initialize_cache(num_gpu_blocks, num_cpu_blocks)
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment