
----------------------------------------------Summary-----------------------------------------------
batch_size_per_gpu: 1
seq_len: 512
tp_size: 1
pp_size: 1
num_tokens_to_generate: 32
use_kv_cache: True
kv_cache_memory_per_gpu: 445.64 MB
kv_cache_latency: 437.12 us
layernorm_dtype_bytes: 2
embedding_memory_per_gpu: 327.68 MB
weight_memory_per_gpu: 25.49 GB
prefill_max_batch_size_per_gpu: 10
prefill_activation_memory_per_gpu: 5.03 GB
prefill_num_flops_fwd_total: 13.27 T
decode_max_batch_size_per_gpu: 120
decode_activation_memory_per_gpu: 6.56 MB
decode_num_flops_fwd_total: 25.49 G
prefill_latency: 43.1 ms
prefill_latency_fwd_attn: 14.45 ms
prefill_latency_fwd_mlp: 27.53 ms
prefill_latency_fwd_layernorm: 411.41 us
prefill_latency_fwd_tp_comm: 0.0 us
prefill_latency_fwd_input_embedding: 160.71 us
prefill_latency_fwd_output_embedding_loss: 537.73 us
decode_latency: 12.94 ms
decode_latency_fwd_attn: 4.12 ms
decode_latency_fwd_mlp: 8.23 ms
decode_latency_fwd_layernorm: 0.8 us
decode_latency_fwd_tp_comm: 0.0 us
decode_latency_fwd_input_embedding: 160.71 us
decode_latency_fwd_output_embedding_loss: 1.05 us
total_decode_latency: 414.23 ms
total_latency: 457.33 ms
total_per_token_latency: 14.29 ms
----------------------------------------------------------------------------------------------------
