Add fields for working GPU count in rollup data and use working GPU count to compute total_throughput_per_gpu - #270
Conversation
count to compute total_throughput_per_gpu Signed-off-by: HuiGao <huig@nvidia.com>
…ount to compute total_throughput_per_gpu
Codecov Report✅ All modified and coverable lines are covered by tests. Additional details and impacted files@@ Coverage Diff @@
## main #270 +/- ##
=======================================
Coverage ? 67.99%
=======================================
Files ? 70
Lines ? 9304
Branches ? 0
=======================================
Hits ? 6326
Misses ? 2978
Partials ? 0 ☔ View full report in Codecov by Harness. 🚀 New features to boost your workflow:
|
|
669968.tar.gz
We can see that the "Total GPU Count" is 32 while the "Total Working GPU Count" is 24 with 4 prefill GPUs sand 20 decode GPUs. This is consistent with config in 669968/config.yaml:
|
We may allocate part of devices on one node, such as 1 prefill server with DEP4 only needs 4 GPUs.
With current formula, this server will be counted with 8 GPUs although it only use 4.
The total throughput per gpu is more accurate to be computed with the real number of working GPUs, so add new logic to get the number of real working GPU devices.