NVIDIA · ko3n1g · Oct 30, 2025 · Oct 30, 2025
diff --git a/...bert/bert_mcore_tp1_pp2/model_config.yaml → ...nal_tests/test_cases/bert/bert_mcore.yaml b/...bert/bert_mcore_tp1_pp2/model_config.yaml → ...nal_tests/test_cases/bert/bert_mcore.yaml
@@ -3,6 +3,7 @@ ENV_VARS:
   NVTE_ALLOW_NONDETERMINISTIC_ALGO: 0
   NCCL_ALGO: Ring
   CUBLAS_WORKSPACE_CONFIG: :4096:8
+TEST_TYPE: ckpt-resume
 MODEL_ARGS:
   --num-layers: 24
   --hidden-size: 1024
@@ -33,12 +34,10 @@ MODEL_ARGS:
   --save-interval: 10000
   --eval-interval: 1000
   --eval-iters: 10
-  --tensor-model-parallel-size: 1
-  --pipeline-model-parallel-size: 2
   --deterministic-mode: true
   --no-gradient-accumulation-fusion: true
   --data-cache-path: ${DATA_CACHE_PATH}
   --bf16: true
   --ckpt-format: torch
   --attention-backend: unfused
-TEST_TYPE: regular
+  --dist-ckpt-strictness: log_all # backward compatibility for TE changes
diff --git a/tests/functional_tests/test_cases/bert/bert_mcore_tp1_pp4_vp2/model_config.yaml b/tests/functional_tests/test_cases/bert/bert_mcore_tp1_pp4_vp2/model_config.yaml
diff --git a/tests/functional_tests/test_cases/bert/bert_mcore_tp2_pp2/model_config.yaml b/tests/functional_tests/test_cases/bert/bert_mcore_tp2_pp2/model_config.yaml
diff --git a/...st_cases/bert/bert_mcore_tp2_pp2_frozen_resume_torch_dist/golden_values_dev_dgx_a100.json b/...st_cases/bert/bert_mcore_tp2_pp2_frozen_resume_torch_dist/golden_values_dev_dgx_a100.json
diff --git a/...st_cases/bert/bert_mcore_tp2_pp2_frozen_resume_torch_dist/golden_values_dev_dgx_h100.json b/...st_cases/bert/bert_mcore_tp2_pp2_frozen_resume_torch_dist/golden_values_dev_dgx_h100.json
diff --git a/...st_cases/bert/bert_mcore_tp2_pp2_frozen_resume_torch_dist/golden_values_lts_dgx_a100.json b/...st_cases/bert/bert_mcore_tp2_pp2_frozen_resume_torch_dist/golden_values_lts_dgx_a100.json
diff --git a/...ional_tests/test_cases/bert/bert_mcore_tp2_pp2_frozen_resume_torch_dist/model_config.yaml b/...ional_tests/test_cases/bert/bert_mcore_tp2_pp2_frozen_resume_torch_dist/model_config.yaml
diff --git a/...ional_tests/test_cases/bert/bert_mcore_tp2_pp2_local_spec/golden_values_dev_dgx_a100.json b/...ional_tests/test_cases/bert/bert_mcore_tp2_pp2_local_spec/golden_values_dev_dgx_a100.json