Skip to content
Merged
Show file tree
Hide file tree
Changes from 2 commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
3 changes: 2 additions & 1 deletion tests/test_utils/python_scripts/generate_jet_trigger_job.py
Original file line number Diff line number Diff line change
Expand Up @@ -225,7 +225,8 @@ def main(
"needs": needs,
"script": [" ".join(script)],
"artifacts": {"paths": ["results/"], "when": "always"},
"allow_failure": test_case["spec"]["model"] == "gpt-nemo",
"allow_failure": test_case["spec"].get("allow_failure", False)
or test_case["spec"]["model"] == "gpt-nemo",
"retry": {
"max": 2,
"when": [
Expand Down
1 change: 1 addition & 0 deletions tests/test_utils/recipes/gb200/gpt.yaml
Original file line number Diff line number Diff line change
Expand Up @@ -208,6 +208,7 @@ products:
- environment: [dev]
scope: [nightly]
platforms: [dgx_gb200]
allow_failure: [true]
- test_case: [gpt3_mcore_te_tp2_pp2_cp2]
products:
- environment: [dev]
Expand Down
2 changes: 2 additions & 0 deletions tests/test_utils/recipes/gb200/moe.yaml
Original file line number Diff line number Diff line change
Expand Up @@ -112,6 +112,7 @@ products:
- environment: [dev]
scope: [nightly]
platforms: [dgx_gb200]
allow_failure: [true]
# - test_case: [gpt3_mcore_te_tp2_pp2_ep4_etp1_mtp_resume_torch_dist_fp8]
# products:
# - environment: [dev]
Expand All @@ -137,6 +138,7 @@ products:
- environment: [dev]
scope: [nightly]
platforms: [dgx_gb200]
allow_failure: [true]
- test_case: [gpt3_mcore_te_tp2_pp1_te_8experts2parallel_overlap_grad_reduce_param_gather_groupedGEMM]
products:
- environment: [dev]
Expand Down
Loading