Clarification on client host memory requirement for 1T model checkpointing – changed between v2.0 and v3.0?
Nobody has claimed this yet.
- Dominant language
- Python
- Stars
- 203
- Forks
- 67
- Avg merge
- 20m
- Merged PRs (30d)
- 8
Description
Hi MLCommons Storage team,
I have a question regarding the client host memory requirement for the checkpointing benchmark, specifically for the 1T model.
According to the current README (https://github.com/mlcommons/storage/tree/main/checkpointing), Table 2 shows that the checkpoint size for the 1T model is 18 TB. The README also states:
"Note: the aggregate simulated accelerator memory across all nodes must be sufficient to accommodate the model’s checkpoint size."
Based on this, I previously understood that the total client host memory across all nodes must be at least 18 TB to run the 1T checkpointing benchmark.
However, when I actually ran the benchmark this year (presumably with the MLPerf Storage v3.0 codebase), I found that the benchmark ran successfully even though the total client host memory was less than 18 TB. This was not possible with the v2.0 codebase, where the benchmark would fail if the memory was insufficient.
This leads me to believe that there may have been a change between v2.0 and v3.0 regarding how the client host memory requirement is enforced. Could you please clarify:
Has the memory requirement been relaxed or removed in v3.0?
If so, what is the new rule or validation logic for client host memory?
Is the --client-host-memory-in-gb parameter still strictly validated against the checkpoint size, or is it now only used for cache clearing decisions (e.g., the "3× memory" rule mentioned in the docs)?
I would appreciate any clarification on this. Thank you!
Contributor guide
First steps
- Read the whole issue, then the project's contributing guide.
- Comment on the issue to say you are picking it up — it saves two people doing the same work.
- Fork the repository and make your change on a branch.
- Open a pull request that references the issue number.
Research direction
Start with the checkpointing README, especially Table 2 and the note about aggregate simulated accelerator memory, then search the v2.0 and v3.0 code for --client-host-memory-in-gb and its validation. Compare the relevant behavior and document whether the requirement changed, including the cache-clearing rule and the conditions that define completion.
Written by the indexing model from the issue text.
Assessment
- Tech stack
- machine-learning
- Domain
- documentation, performance
- Issue type
- Documentation
- Difficulty
- 3/5
- Estimated time
- 1-2 days
- Activity status
- Quiet
- Clarity
- Mostly clear
- Newbie friendliness
- 48/100