Skip to content

Commit 081e905

Browse files
author
zihugithub
committed
[Benchmark] Replace Aquila, DeepSeek and Mixtral benchmark tests with Qwen3
1 parent ca6016e commit 081e905

14 files changed

Lines changed: 50 additions & 335 deletions

File tree

tests/functional_tests/benchmark/aquila/conf/train/data.yaml

Lines changed: 0 additions & 10 deletions
This file was deleted.

tests/functional_tests/benchmark/aquila/conf/train/tp2_pp2.yaml

Lines changed: 0 additions & 62 deletions
This file was deleted.

tests/functional_tests/benchmark/aquila/gold_values/tp2_pp2.json

Lines changed: 0 additions & 16 deletions
This file was deleted.

tests/functional_tests/benchmark/deepseek/conf/tp2_pp2_ep2.yaml

Lines changed: 0 additions & 31 deletions
This file was deleted.

tests/functional_tests/benchmark/deepseek/conf/train/tp2_pp2_ep2.yaml

Lines changed: 0 additions & 105 deletions
This file was deleted.

tests/functional_tests/benchmark/deepseek/gold_values/tp2_pp2_ep2.json

Lines changed: 0 additions & 16 deletions
This file was deleted.

tests/functional_tests/benchmark/mixtral/conf/tp2_pp1_ep2.yaml

Lines changed: 0 additions & 33 deletions
This file was deleted.

tests/functional_tests/benchmark/mixtral/conf/train/data.yaml

Lines changed: 0 additions & 8 deletions
This file was deleted.

tests/functional_tests/benchmark/mixtral/gold_values/tp2_pp1_ep2.json

Lines changed: 0 additions & 16 deletions
This file was deleted.

tests/functional_tests/benchmark/aquila/conf/tp2_pp2.yaml renamed to tests/functional_tests/benchmark/qwen3/conf/tp2_pp2.yaml

Lines changed: 1 addition & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -4,7 +4,7 @@ defaults:
44

55
experiment:
66
exp_name: tp2_pp2
7-
exp_dir: tests/functional_tests/benchmark/aquila/test_results/tp2_pp2
7+
exp_dir: tests/functional_tests/benchmark/qwen3/test_results/tp2_pp2
88
task:
99
type: train
1010
backend: megatron
@@ -20,8 +20,6 @@ experiment:
2020
NCCL_ALGO: "Ring"
2121
NVTE_APPLY_QK_LAYER_SCALING: 0
2222
NVTE_ALLOW_NONDETERMINISTIC_ALGO: 0
23-
NVTE_FLASH_ATTN: 0
24-
NVTE_FUSED_ATTN: 0
2523
CUDNN_BENCHMARK: "false"
2624
CUDNN_DETERMINISTIC: "true"
2725
cmds:

0 commit comments

Comments
 (0)