huggingface/text-generation-inference
5
0
Fork 0
mirror of https://github.com/huggingface/text-generation-inference.git synced 2025-06-18 23:32:06 +00:00
Code Issues Packages Projects Releases Wiki Activity

Add log_master & VLM cases

Browse Source
This commit is contained in:
regisss 2025-06-17 21:13:13 +00:00
parent 564c9e1cc0
commit 9dbaa176fd
3 changed files with 18 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
backends/gaudi/server/text_generation_server/models/flash_causal_lm.py
View File

@ -1754,10 +1754,11 @@ class FlashCausalLM(Model):
total_batch_seq = 0.001
total_mem = 0
available_mem = prompt_available_memory
logger.info(
msg = (
f"Prefill batch size list:{[bsz[0] for bsz in buckets]}\n"
f"Prefill sequence length list:{[seq[1] for seq in buckets]}\n"
)
log_master(logger.info, msg)
for i, (batch_size, seq_len) in enumerate(buckets):
if batch_size * seq_len > self.max_batch_prefill_tokens:
continue
@ -1784,7 +1785,7 @@ class FlashCausalLM(Model):
total_mem += used_mem
total_batch_seq += batch_seq
logger.info("Prefill warmup successful.\n")
log_master(logger.info, "Prefill warmup successful.\n")
def ordering_function_max_bs(b):
return (-b[0], b[1])
@ -1797,7 +1798,7 @@ class FlashCausalLM(Model):
total_batch_seq = 0.001
total_mem = 0
available_mem = free_mem - self.mem_reserved
logger.info(f"Decode batch size list:{[bsz[0] for bsz in buckets]}\n")
log_master(logger.info, f"Decode batch size list:{[bsz[0] for bsz in buckets]}\n")
for i, (batch_size, block_num) in enumerate(buckets):
if batch_size > block_num:
continue
@ -1822,7 +1823,7 @@ class FlashCausalLM(Model):
total_mem += used_mem
total_batch_seq += batch_seq
logger.info("Decode warmup successful.\n")
log_master(logger.info, "Decode warmup successful.\n")
log_master(
logger.info,

3
backends/gaudi/server/text_generation_server/models/flash_vlm_causal_lm.py
View File

@ -822,6 +822,7 @@ class FlashVlmCausalLM(FlashCausalLM):
total_batch_seq = 0.001
total_mem = 0
available_mem = decode_available_memory
log_master(logger.info, f"Decode batch size list:{[bsz[0] for bsz in buckets]}\n")
for i, (batch_size, block_num) in enumerate(buckets):
if batch_size > block_num:
continue
@ -847,6 +848,8 @@ class FlashVlmCausalLM(FlashCausalLM):
total_mem += used_mem
total_batch_seq += batch_seq
log_master(logger.info, "Decode warmup successful.\n")
log_master(
logger.info,
f"warmup hpu graph time {int(time.time() - start_time)}s warmup shape count {warmup_shape_count}",

10
backends/gaudi/server/text_generation_server/models/mllama_causal_lm.py
View File

@ -398,6 +398,11 @@ class FlashMllamaCausalLM(FlashVlmCausalLM):
total_batch_seq = 0.001
total_mem = 0
available_mem = prompt_available_memory
msg = (
f"Prefill batch size list:{[bsz[0] for bsz in buckets]}\n"
f"Prefill sequence length list:{[seq[1] for seq in buckets]}\n"
)
log_master(logger.info, msg)
for i, (batch_size, seq_len) in enumerate(buckets):
if batch_size * seq_len > self.max_batch_prefill_tokens:
continue
@ -424,6 +429,8 @@ class FlashMllamaCausalLM(FlashVlmCausalLM):
total_mem += used_mem
total_batch_seq += batch_seq
log_master(logger.info, "Prefill warmup successful.\n")
def ordering_function_max_bs(b):
return (-b[0], b[1])
@ -435,6 +442,7 @@ class FlashMllamaCausalLM(FlashVlmCausalLM):
total_batch_seq = 0.001
total_mem = 0
available_mem = free_mem - self.mem_reserved
log_master(logger.info, f"Decode batch size list:{[bsz[0] for bsz in buckets]}\n")
for i, (batch_size, block_num) in enumerate(buckets):
if batch_size > block_num:
continue
@ -459,6 +467,8 @@ class FlashMllamaCausalLM(FlashVlmCausalLM):
total_mem += used_mem
total_batch_seq += batch_seq
log_master(logger.info, "Decode warmup successful.\n")
log_master(
logger.info,
f"warmup hpu graph time {int(time.time() - start_time)}s warmup shape count {warmup_shape_count}",