Skip to content

Commit e6e040b

Browse files
Rename benchmark: gnn -> rgat
1 parent 02de5ad commit e6e040b

File tree

10 files changed

+23
-18
lines changed

10 files changed

+23
-18
lines changed

mlperf_logging/benchmark_meta.py

Lines changed: 4 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -10,12 +10,14 @@
1010
'minigo': 10,
1111
'resnet': 5,
1212
'ssd': 5,
13+
'retinanet': 5,
1314
'stable_diffusion': 10,
1415
'transformer': 10,
1516
'ncf': 10,
1617
'rnnt': 10,
1718
'unet3d': 40,
18-
'gnn' : 10,
19+
'gnn' : 10,
20+
'rgat': 10,
1921
'llama2_70b_lora': 10,
2022
},
2123

@@ -138,7 +140,7 @@
138140
'retinanet',
139141
'stable_diffusion',
140142
'llama2_70b_lora',
141-
'gnn'
143+
'rgat'
142144
]
143145
},
144146

mlperf_logging/compliance_checker/README.md

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -28,13 +28,13 @@ As log examples use [NVIDIA's training logs](https://github.com/mlperf/training_
2828
5.0.0/closed_retinanet.yaml - Per-benchmark rules, closed submissions.
2929
5.0.0/closed_bert.yaml
3030
5.0.0/closed_dlrm_dcnv2.yaml
31-
5.0.0/closed_gnn.yaml
31+
5.0.0/closed_rgat.yaml
3232
5.0.0/closed_llama2_70b_lora.yaml
3333
5.0.0/closed_stable_diffusion.yaml
3434
5.0.0/open_retinanet.yaml - Per-benchmark rules, open submissions.
3535
5.0.0/open_bert.yaml
3636
5.0.0/open_dlrm_dcnv2.yaml
37-
5.0.0/open_gnn.yaml
37+
5.0.0/open_rgat.yaml
3838
5.0.0/open_llama2_70b_lora.yaml
3939
5.0.0/open_stable_diffusion.yaml
4040

mlperf_logging/compliance_checker/training_5.0.0/closed_common.yaml

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -2,7 +2,7 @@
22
- KEY:
33
NAME: submission_benchmark
44
REQ: EXACTLY_ONE
5-
CHECK: " v['value'] in ['retinanet', 'stable_diffusion', 'dlrm_dcnv2', 'bert', 'gnn', 'llama2_70b_lora'] "
5+
CHECK: " v['value'] in ['retinanet', 'stable_diffusion', 'dlrm_dcnv2', 'bert', 'rgat', 'llama2_70b_lora'] "
66
POST: " enqueue_config('training_4.1.0/closed_{}.yaml'.format(v['value'])) "
77

88
- KEY:

mlperf_logging/compliance_checker/training_5.0.0/open_common.yaml

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -2,5 +2,5 @@
22
- KEY:
33
NAME: submission_benchmark
44
REQ: EXACTLY_ONE
5-
CHECK: " v['value'] in ['retinanet', 'stable_diffusion', 'dlrm_dcnv2', 'bert', 'gnn', 'llama2_70b_lora'] "
5+
CHECK: " v['value'] in ['retinanet', 'stable_diffusion', 'dlrm_dcnv2', 'bert', 'rgat', 'llama2_70b_lora'] "
66
POST: " enqueue_config('training_4.1.0/open_{}.yaml'.format(v['value'])) "

mlperf_logging/mllog/constants.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -53,6 +53,7 @@
5353
GPT3 = "gpt3"
5454
LLAMA2_70B_LORA = "llama2_70b_lora"
5555
GNN = "gnn"
56+
RGAT = "rgat"
5657

5758
# Constant values - model info
5859
ADAGRAD = "adagrad"

mlperf_logging/rcp_checker/rcp_checker.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -25,10 +25,12 @@
2525
'maskrcnn' : 5,
2626
'resnet' : 5,
2727
'ssd' : 5,
28+
'retinanet': 5,
2829
'unet3d' : 40,
2930
'rnnt': 10,
3031
'stable_diffusion': 10,
31-
'gnn': 10,
32+
'gnn': 10,
33+
'rgat': 10,
3234
'llama2_70b_lora': 10,
3335
},
3436
"hpc": {

mlperf_logging/rcp_checker/training_5.0.0/rcps_gnn.json renamed to mlperf_logging/rcp_checker/training_5.0.0/rcps_rgat.json

Lines changed: 10 additions & 10 deletions
Original file line numberDiff line numberDiff line change
@@ -1,8 +1,8 @@
11
{
22

3-
"gnn_ref_4096":
3+
"rgat_ref_4096":
44
{
5-
"Benchmark": "gnn",
5+
"Benchmark": "rgat",
66
"Creator": "NVIDIA",
77
"When": "Reference RCPs before v4.0",
88
"Platform": "1xDGX-A100 and 8xDGX-A100",
@@ -17,9 +17,9 @@
1717
0.80,0.85 ]
1818
},
1919

20-
"gnn_ref_16384":
20+
"rgat_ref_16384":
2121
{
22-
"Benchmark": "gnn",
22+
"Benchmark": "rgat",
2323
"Creator": "NVIDIA",
2424
"When": "Reference RCPs before v4.0",
2525
"Platform": "8xDGX-A100",
@@ -34,9 +34,9 @@
3434
0.85,0.90 ]
3535
},
3636

37-
"gnn_ref_32768":
37+
"rgat_ref_32768":
3838
{
39-
"Benchmark": "gnn",
39+
"Benchmark": "rgat",
4040
"Creator": "Intel",
4141
"When": "Reference RCPs before v4.0",
4242
"Platform": "16xSPR-2S",
@@ -51,9 +51,9 @@
5151
0.90,0.90 ]
5252
},
5353

54-
"gnn_ref_65536":
54+
"rgat_ref_65536":
5555
{
56-
"Benchmark": "gnn",
56+
"Benchmark": "rgat",
5757
"Creator": "NVIDIA",
5858
"When": "Reference RCPs before v4.0",
5959
"Platform": "32xDGX-A100",
@@ -69,9 +69,9 @@
6969
]
7070
},
7171

72-
"gnn_ref_262144":
72+
"rgat_ref_262144":
7373
{
74-
"Benchmark": "gnn",
74+
"Benchmark": "rgat",
7575
"Creator": "NVIDIA",
7676
"When": "Reference RCPs before v4.0",
7777
"Platform": "128xDGX-H100",

mlperf_logging/result_summarizer/config.yaml

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -90,7 +90,7 @@ columns:
9090
retinanet: ["Benchmark results (minutes)", "Object detection, light-weight", "OpenImages", "RetinaNet"]
9191
stable_diffusion: ["Benchmark results (minutes)", "Text to image", "Laion 400m and Coco-2017", "StableDiffusion"]
9292
llama2_70b_lora: ["Benchmark results (minutes)", "LLM-Finetune", "SCROLSS Gov Report", "LLama2-70B-LoRA"]
93-
gnn: ["Benchmark results (minutes)", "Graph node classification", "IGBH-Full", "R-GAT"]
93+
rgat: ["Benchmark results (minutes)", "Graph node classification", "IGBH-Full", "R-GAT"]
9494
default: [" ", " ", " "]
9595

9696
hpc:

0 commit comments

Comments
 (0)