Skip to content

Commit

Permalink
Flaky test optional until fixed (NVIDIA#10495)
Browse files Browse the repository at this point in the history
  • Loading branch information
pablo-garay authored Sep 16, 2024
1 parent d419955 commit 16568d7
Showing 1 changed file with 4 additions and 3 deletions.
7 changes: 4 additions & 3 deletions .github/workflows/cicd-main.yml
Original file line number Diff line number Diff line change
Expand Up @@ -3466,10 +3466,10 @@ jobs:
rm -rf examples/nlp/language_modeling/gpt_pretrain_results
rm -rf examples/nlp/language_modeling/gpt_index_mappings
L2_Megatron_GPT_Auto_Configurator_TP1_PP1_MBS124:
OPTIONAL_L2_Megatron_GPT_Auto_Configurator_TP1_PP1_MBS124:
needs: [cicd-test-container-setup]
uses: ./.github/workflows/_test_template.yml
if: contains(fromJSON(needs.cicd-test-container-setup.outputs.test_to_run), 'L2_Megatron_GPT_Auto_Configurator_TP1_PP1_MBS124') || needs.cicd-test-container-setup.outputs.all == 'true'
if: contains(fromJSON(needs.cicd-test-container-setup.outputs.test_to_run), 'OPTIONAL_L2_Megatron_GPT_Auto_Configurator_TP1_PP1_MBS124') || needs.cicd-test-container-setup.outputs.all == 'true'
with:
RUNNER: self-hosted-azure-gpus-1
SCRIPT: |
Expand Down Expand Up @@ -3500,6 +3500,7 @@ jobs:
--get_results
AFTER_SCRIPT: |
rm -rf examples/llm/auto_configurator/auto_conf_logs
IS_OPTIONAL: true

L2_Megatron_GPT_Finetuning_PP2:
needs: [cicd-test-container-setup]
Expand Down Expand Up @@ -5252,7 +5253,7 @@ jobs:
- L2_Megatron_GPT_with_ALiBi_Pretraining_and_Resume_Training_TP2
- L2_Megatron_GPT_with_KERPLE_Pretraining_and_Resume_Training_TP2
- L2_Megatron_GPT_Pretraining_and_Resume_Training_PP2
- L2_Megatron_GPT_Auto_Configurator_TP1_PP1_MBS124
#- OPTIONAL_L2_Megatron_GPT_Auto_Configurator_TP1_PP1_MBS124
- L2_Megatron_GPT_Finetuning_PP2
- L2_Megatron_GPT_Finetuning_StarCoder_PP1
- L2_Megatron_GPT_Embedding
Expand Down

0 comments on commit 16568d7

Please sign in to comment.