Skip to content

Commit b8b3adc

Browse files
Retire llama3.1 405B and DLRM-DCNv2 references for the v6.1 round. (#901)
Move the retired implementations under retired_benchmarks, add DLRMv4 to the v6.1 README table, and point historical README links at the new paths. Co-authored-by: Cursor <cursoragent@cursor.com>
1 parent 685d094 commit b8b3adc

37 files changed

Lines changed: 10 additions & 9 deletions

‎README.md‎

Lines changed: 9 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -52,6 +52,7 @@ Some these benchmarks are rather slow or take a long time to run on the referenc
5252
| gpt_oss_20b | [small_llm_moe_pretraining](https://github.com/mlcommons/training/tree/master/small_llm_moe_pretraining/primus) | Primus | C4 | 20B |
5353
| deepseekv3 | [llm_moe_pretraining](https://github.com/mlcommons/training/tree/master/llm_moe_pretraining/nemo) | NeMo | C4 | 671B |
5454
| qwen35_397b_grpo | [llm_post_training](https://github.com/mlcommons/training/tree/master/llm_post_training) | NeMo-RL / NeMo-Gym | SWE tasks | 397B
55+
| dlrm_v4_hstu | [recommendation](https://github.com/mlcommons/training/tree/master/recommendation) | pytorch | Yambda-5b | HSTU
5556

5657
*Framework here is given for the reference implementation. Submitters are free to use their own frameworks to run the benchmark.
5758

@@ -64,8 +65,8 @@ Some these benchmarks are rather slow or take a long time to run on the referenc
6465
| flux.1 | [text_to_image](https://github.com/mlcommons/training/tree/master/text_to_image) | torchtitan | CC12M subset | 11.9B
6566
| llama3.1_8b | [small_llm_pretraining](https://github.com/mlcommons/training/tree/master/small_llm_pretraining) | NeMo | C4 | 8b
6667
| llama2_70b_lora | [llama2_70b_lora](https://github.com/mlcommons/training/tree/master/llama2_70b_lora) | pytorch | SCROLLS GovReport | 70B
67-
| llama3.1_405b | [large_language_model_pretraining](https://github.com/mlcommons/training/tree/master/large_language_model_pretraining) | NeMo | C4 | 405B
68-
| dlrm_dcnv2 | [recommendation_v2](https://github.com/mlcommons/training/tree/master/recommendation_v2/torchrec_dlrm) | torchrec | Criteo 3.5TB multi-hot | 167M
68+
| llama3.1_405b | [large_language_model_pretraining](https://github.com/mlcommons/training/tree/master/retired_benchmarks/large_language_model_pretraining) | NeMo | C4 | 405B
69+
| dlrm_dcnv2 | [recommendation_v2](https://github.com/mlcommons/training/tree/master/retired_benchmarks/recommendation_v2/torchrec_dlrm) | torchrec | Criteo 3.5TB multi-hot | 167M
6970
| gpt_oss_20b | [small_llm_moe_pretraining](https://github.com/mlcommons/training/tree/master/small_llm_moe_pretraining/primus) | Primus | C4 | 20B |
7071
| deepseekv3 | [llm_moe_pretraining](https://github.com/mlcommons/training/tree/master/llm_moe_pretraining/nemo) | NeMo | C4 | 671B |
7172

@@ -81,8 +82,8 @@ Some these benchmarks are rather slow or take a long time to run on the referenc
8182
| flux.1 | [text_to_image](https://github.com/mlcommons/training/tree/master/text_to_image) | torchtitan | CC12M subset | 11.9B
8283
| llama3.1_8b | [small_llm_pretraining](https://github.com/mlcommons/training/tree/master/small_llm_pretraining) | NeMo | C4 | 8b
8384
| llama2_70b_lora | [llama2_70b_lora](https://github.com/mlcommons/training/tree/master/llama2_70b_lora) | pytorch | SCROLLS GovReport | 70B
84-
| llama3.1_405b | [large_language_model_pretraining](https://github.com/mlcommons/training/tree/master/large_language_model_pretraining) | NeMo | C4 | 405B
85-
| dlrm_dcnv2 | [recommendation_v2](https://github.com/mlcommons/training/tree/master/recommendation_v2/torchrec_dlrm) | torchrec | Criteo 3.5TB multi-hot | 167M
85+
| llama3.1_405b | [large_language_model_pretraining](https://github.com/mlcommons/training/tree/master/retired_benchmarks/large_language_model_pretraining) | NeMo | C4 | 405B
86+
| dlrm_dcnv2 | [recommendation_v2](https://github.com/mlcommons/training/tree/master/retired_benchmarks/recommendation_v2/torchrec_dlrm) | torchrec | Criteo 3.5TB multi-hot | 167M
8687
| rgat | [graph_neural_network](https://github.com/mlcommons/training/tree/master/graph_neural_network) | GLT | IGBH-Full | 25M
8788

8889
*Framework here is given for the reference implementation. Submitters are free to use their own frameworks to run the benchmark.
@@ -99,9 +100,9 @@ Some these benchmarks are rather slow or take a long time to run on the referenc
99100
| retinanet | [single_stage_detector](https://github.com/mlcommons/training/tree/master/single_stage_detector) | pytorch | OpenImages | 37M
100101
| Stable Diffusionv2 | [image generation](https://github.com/mlcommons/training/tree/master/stable_diffusion) | pytorch | LAION-400M-filtered | 865M
101102
| bert | [language_model](https://github.com/mlcommons/training/tree/master/language_model/tensorflow/bert) | tensorflow | Wikipedia 2020/01/01 | 340M
102-
| llama3.1_405b | [large_language_model_pretraining](https://github.com/mlcommons/training/tree/master/large_language_model_pretraining) | NeMo | C4 | 405B
103+
| llama3.1_405b | [large_language_model_pretraining](https://github.com/mlcommons/training/tree/master/retired_benchmarks/large_language_model_pretraining) | NeMo | C4 | 405B
103104
| llama2_70b_lora | [llama2_70b_lora](https://github.com/mlcommons/training/tree/master/llama2_70b_lora) | pytorch | SCROLLS GovReport | 70B
104-
| dlrm_dcnv2 | [recommendation_v2](https://github.com/mlcommons/training/tree/master/recommendation_v2/torchrec_dlrm) | torchrec | Criteo 3.5TB multi-hot | 167M
105+
| dlrm_dcnv2 | [recommendation_v2](https://github.com/mlcommons/training/tree/master/retired_benchmarks/recommendation_v2/torchrec_dlrm) | torchrec | Criteo 3.5TB multi-hot | 167M
105106
| rgat | [graph_neural_network](https://github.com/mlcommons/training/tree/master/graph_neural_network) | GLT | IGBH-Full | 25M
106107

107108
# MLPerf Training v4.1 (Submission Deadline Oct 11, 2024)
@@ -114,7 +115,7 @@ Some these benchmarks are rather slow or take a long time to run on the referenc
114115
| BERT-large | [language/nlp](https://github.com/mlcommons/training/tree/master/language_model/tensorflow/bert) | tensorflow | Wikipedia 2020/01/01 | 340M
115116
| GPT3 | [language/llm](https://github.com/mlcommons/training/tree/master/large_language_model) | paxml,megatron-lm | C4 | 175B
116117
| LLama2 70B-LoRA | [language/LLM fine-tuning](https://github.com/mlcommons/training/tree/master/llama2_70b_lora) | pytorch | SCROLLS GovReport | 70B
117-
| DLRMv2 | [recommendation](https://github.com/mlcommons/training/tree/master/recommendation_v2/torchrec_dlrm) | torchrec | Criteo 3.5TB multi-hot | 167M
118+
| DLRMv2 | [recommendation](https://github.com/mlcommons/training/tree/master/retired_benchmarks/recommendation_v2/torchrec_dlrm) | torchrec | Criteo 3.5TB multi-hot | 167M
118119
| RGAT | [GNN](https://github.com/mlcommons/training/tree/master/graph_neural_network) | pytorch | IGBH-Full | 25M
119120

120121
*Note model parameter count is not the same as active parameter that are being trained in the benchmark.
@@ -131,7 +132,7 @@ Some these benchmarks are rather slow or take a long time to run on the referenc
131132
| BERT-large | [language/nlp](https://github.com/mlcommons/training/tree/master/language_model/tensorflow/bert) | tensorflow | Wikipedia 2020/01/01
132133
| GPT3 | [language/llm](https://github.com/mlcommons/training/tree/master/large_language_model) | paxml,megatron-lm | C4
133134
| LLama2 70B-LoRA | [language/LLM fine-tuning](https://github.com/mlcommons/training/tree/master/llama2_70b_lora) | pytorch | SCROLLS GovReport
134-
| DLRMv2 | [recommendation](https://github.com/mlcommons/training/tree/master/recommendation_v2/torchrec_dlrm) | torchrec | Criteo 3.5TB multi-hot
135+
| DLRMv2 | [recommendation](https://github.com/mlcommons/training/tree/master/retired_benchmarks/recommendation_v2/torchrec_dlrm) | torchrec | Criteo 3.5TB multi-hot
135136
| RGAT | [GNN](https://github.com/mlcommons/training/tree/master/graph_neural_network) | pytorch | IGBFull
136137

137138

large_language_model_pretraining/nemo/.dockerignore renamed to retired_benchmarks/large_language_model_pretraining/nemo/.dockerignore

File renamed without changes.

large_language_model_pretraining/nemo/Dockerfile renamed to retired_benchmarks/large_language_model_pretraining/nemo/Dockerfile

File renamed without changes.

large_language_model_pretraining/nemo/Dockerfile_mlcube renamed to retired_benchmarks/large_language_model_pretraining/nemo/Dockerfile_mlcube

File renamed without changes.

large_language_model_pretraining/nemo/README.md renamed to retired_benchmarks/large_language_model_pretraining/nemo/README.md

File renamed without changes.

large_language_model_pretraining/nemo/callbacks.py renamed to retired_benchmarks/large_language_model_pretraining/nemo/callbacks.py

File renamed without changes.

large_language_model_pretraining/nemo/config.sh renamed to retired_benchmarks/large_language_model_pretraining/nemo/config.sh

File renamed without changes.

large_language_model_pretraining/nemo/demo_config.sh renamed to retired_benchmarks/large_language_model_pretraining/nemo/demo_config.sh

File renamed without changes.

large_language_model_pretraining/nemo/mcore.patch renamed to retired_benchmarks/large_language_model_pretraining/nemo/mcore.patch

File renamed without changes.

large_language_model_pretraining/nemo/mlcube/.gitignore renamed to retired_benchmarks/large_language_model_pretraining/nemo/mlcube/.gitignore

File renamed without changes.

0 commit comments

Comments
 (0)