verl/utils/flops_counter.py

• get_device_flops · function · L21-L48 — def get_device_flops(unit="T")
• unit_convert · function · L23-L31 — def unit_convert(number, level)
• FlopsCounter · class · L51-L123 — class FlopsCounter
• init · method · L61-L67 — def init(self, config: PretrainedConfig)
• _estimate_unknown_flops · method · L69-L70 — def _estimate_unknown_flops(self, tokens_sum, batch_seqlens, delta_time)
• _estimate_qwen2_flops · method · L72-L105 — def _estimate_qwen2_flops(self, tokens_sum, batch_seqlens, delta_time)
• estimate_flops · method · L107-L123 — def estimate_flops(self, batch_seqlens, delta_time)