Skip to content
Merged
Show file tree
Hide file tree
Changes from 1 commit
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
The table of contents is too big for display.
Diff view
Diff view
  •  
  •  
  •  
Original file line number Diff line number Diff line change
Expand Up @@ -42,3 +42,4 @@ MODEL_ARGS:
--ckpt-format: torch
--attention-backend: unfused
TEST_TYPE: regular
LAUNCHER: ft_launcher
Original file line number Diff line number Diff line change
Expand Up @@ -43,3 +43,4 @@ MODEL_ARGS:
--ckpt-format: torch
--attention-backend: unfused
TEST_TYPE: regular
LAUNCHER: ft_launcher
Original file line number Diff line number Diff line change
Expand Up @@ -42,3 +42,4 @@ MODEL_ARGS:
--ckpt-format: torch
--attention-backend: unfused
TEST_TYPE: regular
LAUNCHER: ft_launcher
Original file line number Diff line number Diff line change
Expand Up @@ -45,3 +45,4 @@ MODEL_ARGS:
--dist-ckpt-strictness: log_all # backward compatibility for TE changes
--attention-backend: unfused
TEST_TYPE: frozen-resume
LAUNCHER: ft_launcher
Original file line number Diff line number Diff line change
Expand Up @@ -43,3 +43,4 @@ MODEL_ARGS:
--ckpt-format: torch
--attention-backend: local
TEST_TYPE: regular
LAUNCHER: ft_launcher
Original file line number Diff line number Diff line change
Expand Up @@ -45,3 +45,4 @@ MODEL_ARGS:
--dist-ckpt-strictness: log_all # backward compatibility for TE changes
--attention-backend: unfused
TEST_TYPE: ckpt-resume
LAUNCHER: ft_launcher
Original file line number Diff line number Diff line change
Expand Up @@ -46,3 +46,4 @@ MODEL_ARGS:
--ckpt-format: torch
--attention-backend: local
TEST_TYPE: ckpt-resume
LAUNCHER: ft_launcher
Original file line number Diff line number Diff line change
Expand Up @@ -42,3 +42,4 @@ MODEL_ARGS:
--ckpt-format: torch
--attention-backend: unfused
TEST_TYPE: regular
LAUNCHER: ft_launcher
Original file line number Diff line number Diff line change
Expand Up @@ -5,3 +5,4 @@ ENV_VARS:
CUBLAS_WORKSPACE_CONFIG: :4096:8
MODEL_ARGS:
TEST_TYPE: regular
LAUNCHER: ft_launcher
Original file line number Diff line number Diff line change
Expand Up @@ -15,3 +15,4 @@ MODEL_ARGS:
data.seq_length: 512
log.log_dir: ${CHECKPOINT_SAVE_PATH}
TEST_TYPE: regular
LAUNCHER: ft_launcher
Original file line number Diff line number Diff line change
Expand Up @@ -15,3 +15,4 @@ MODEL_ARGS:
data.global_batch_size: 8
data.seq_length: 2048
TEST_TYPE: regular
LAUNCHER: ft_launcher
Original file line number Diff line number Diff line change
Expand Up @@ -30,3 +30,4 @@ MODEL_ARGS:
data.seq_length: 2048
log.log_dir: ${CHECKPOINT_SAVE_PATH}
TEST_TYPE: regular
LAUNCHER: ft_launcher
Original file line number Diff line number Diff line change
Expand Up @@ -20,3 +20,4 @@ MODEL_ARGS:
data.seq_length: 2048
log.log_dir: ${CHECKPOINT_SAVE_PATH}
TEST_TYPE: regular
LAUNCHER: ft_launcher
Original file line number Diff line number Diff line change
Expand Up @@ -19,3 +19,4 @@ MODEL_ARGS:
data.global_batch_size: 8
data.seq_length: 2048
TEST_TYPE: regular
LAUNCHER: ft_launcher
Original file line number Diff line number Diff line change
Expand Up @@ -13,3 +13,4 @@ MODEL_ARGS:
data.global_batch_size: 64
data.seq_length: 512
TEST_TYPE: regular
LAUNCHER: ft_launcher
Original file line number Diff line number Diff line change
Expand Up @@ -67,3 +67,4 @@ METRICS:
- "num-zeros"
- "mem-allocated-bytes"
- "mem-max-allocated-bytes"
LAUNCHER: ft_launcher
Original file line number Diff line number Diff line change
Expand Up @@ -65,3 +65,4 @@ METRICS:
- "num-zeros"
- "mem-allocated-bytes"
- "mem-max-allocated-bytes"
LAUNCHER: ft_launcher
Original file line number Diff line number Diff line change
Expand Up @@ -85,4 +85,5 @@ AFTER_SCRIPT: |
check_log_not -F "WARNING:megatron.core.rerun_state_machine:Result validation enabled"
check_log -F "Setting rerun_state_machine.current_iteration to 0..."
EXIT_CODE=0
TEST_TYPE: regular
TEST_TYPE: regular
LAUNCHER: ft_launcher
Original file line number Diff line number Diff line change
Expand Up @@ -82,4 +82,5 @@ AFTER_SCRIPT: |
check_log() { if [[ -z $(grep -r $1 "$2" $LOG_DIR) ]]; then exit 1; else echo OK; fi }
check_log -F "WARNING:megatron.core.rerun_state_machine:Result validation enabled"
EXIT_CODE=0
TEST_TYPE: regular
TEST_TYPE: regular
LAUNCHER: ft_launcher
Original file line number Diff line number Diff line change
Expand Up @@ -88,3 +88,4 @@ AFTER_SCRIPT: |
check_log -F "Saving a checkpoint and exiting now. Please resume the job from the checkpoint to rerun the last iteration and establish a diagnostic"
EXIT_CODE=0
TEST_TYPE: regular
LAUNCHER: ft_launcher
Original file line number Diff line number Diff line change
Expand Up @@ -88,3 +88,4 @@ AFTER_SCRIPT: |
check_log -F "Saving a checkpoint and exiting now. Please resume the job from the checkpoint to rerun the last iteration and establish a diagnostic"
EXIT_CODE=0
TEST_TYPE: regular
LAUNCHER: ft_launcher
Original file line number Diff line number Diff line change
Expand Up @@ -83,4 +83,5 @@ AFTER_SCRIPT: |
check_log -F "WARNING:megatron.core.rerun_state_machine:Result validation enabled"
check_log -E "ERROR:megatron\.core\.rerun_state_machine:Rank [0-9]+, node ([0-9a-z]|\-)+, device [0-9]+: Possible persistent error!!"
EXIT_CODE=0
TEST_TYPE: frozen-start
TEST_TYPE: frozen-start
LAUNCHER: ft_launcher
Original file line number Diff line number Diff line change
Expand Up @@ -84,4 +84,5 @@ AFTER_SCRIPT: |
check_log -F "WARNING:megatron.core.rerun_state_machine:Result validation enabled"
check_log -F "Job sharding has changed: Rerun state will be ignored"
EXIT_CODE=0
TEST_TYPE: frozen-start
TEST_TYPE: frozen-start
LAUNCHER: ft_launcher
Original file line number Diff line number Diff line change
Expand Up @@ -83,4 +83,5 @@ AFTER_SCRIPT: |
check_log -F "successfully loaded checkpoint"
check_log -F "WARNING:megatron.core.rerun_state_machine:Result validation enabled"
EXIT_CODE=0
TEST_TYPE: frozen-start
TEST_TYPE: frozen-start
LAUNCHER: ft_launcher
Original file line number Diff line number Diff line change
Expand Up @@ -151,3 +151,4 @@ MODEL_ARGS_5:
--tensor-model-parallel-size: 1
--context-parallel-size: 1
--pipeline-model-parallel-size: 2
LAUNCHER: ft_launcher
Original file line number Diff line number Diff line change
Expand Up @@ -88,3 +88,4 @@ AFTER_SCRIPT: |
check_log -E "ERROR:megatron\.core\.rerun_state_machine:Rank [0-9]+, node ([0-9a-z]|\-)+, device [0-9]+: Possible transient error!!"
EXIT_CODE=0
TEST_TYPE: regular
LAUNCHER: ft_launcher
Original file line number Diff line number Diff line change
Expand Up @@ -56,3 +56,4 @@ MODEL_ARGS:
--async-save: true
--use-persistent-ckpt-worker: true
TEST_TYPE: regular
LAUNCHER: ft_launcher
Original file line number Diff line number Diff line change
Expand Up @@ -56,3 +56,4 @@ MODEL_ARGS:
--async-save: true
--use-persistent-ckpt-worker: true
TEST_TYPE: regular
LAUNCHER: ft_launcher
Original file line number Diff line number Diff line change
Expand Up @@ -54,3 +54,4 @@ MODEL_ARGS:
--async-save: true
--use-persistent-ckpt-worker: true
TEST_TYPE: regular
LAUNCHER: ft_launcher
Original file line number Diff line number Diff line change
Expand Up @@ -54,3 +54,4 @@ MODEL_ARGS:
--async-save: true
--use-persistent-ckpt-worker: true
TEST_TYPE: regular
LAUNCHER: ft_launcher
Original file line number Diff line number Diff line change
Expand Up @@ -54,3 +54,4 @@ MODEL_ARGS:
--log-memory-to-tensorboard: true
--use-persistent-ckpt-worker: true
TEST_TYPE: frozen-resume
LAUNCHER: ft_launcher
Original file line number Diff line number Diff line change
Expand Up @@ -57,3 +57,4 @@ MODEL_ARGS:
--async-save: true
--use-persistent-ckpt-worker: true
TEST_TYPE: regular
LAUNCHER: ft_launcher
Original file line number Diff line number Diff line change
Expand Up @@ -54,3 +54,4 @@ MODEL_ARGS:
--log-memory-to-tensorboard: true
--use-persistent-ckpt-worker: true
TEST_TYPE: ckpt-resume
LAUNCHER: ft_launcher
Original file line number Diff line number Diff line change
Expand Up @@ -54,3 +54,4 @@ MODEL_ARGS:
--log-memory-to-tensorboard: true
--use-persistent-ckpt-worker: true
TEST_TYPE: ckpt-resume
LAUNCHER: ft_launcher
Original file line number Diff line number Diff line change
Expand Up @@ -55,3 +55,4 @@ MODEL_ARGS:
--async-save: true
--use-persistent-ckpt-worker: true
TEST_TYPE: ckpt-resume
LAUNCHER: ft_launcher
Original file line number Diff line number Diff line change
Expand Up @@ -55,3 +55,4 @@ MODEL_ARGS:
--async-save: true
--use-persistent-ckpt-worker: true
TEST_TYPE: ckpt-resume
LAUNCHER: ft_launcher
Original file line number Diff line number Diff line change
Expand Up @@ -55,3 +55,4 @@ MODEL_ARGS:
--async-save: true
--use-persistent-ckpt-worker: true
TEST_TYPE: ckpt-resume
LAUNCHER: ft_launcher
Original file line number Diff line number Diff line change
Expand Up @@ -56,3 +56,4 @@ MODEL_ARGS:
--async-save: true
--use-persistent-ckpt-worker: true
TEST_TYPE: regular
LAUNCHER: ft_launcher
Original file line number Diff line number Diff line change
Expand Up @@ -55,3 +55,4 @@ MODEL_ARGS:
--async-save: true
--use-persistent-ckpt-worker: true
TEST_TYPE: regular
LAUNCHER: ft_launcher
Original file line number Diff line number Diff line change
Expand Up @@ -56,3 +56,4 @@ MODEL_ARGS:
--async-save: true
--use-persistent-ckpt-worker: true
TEST_TYPE: ckpt-resume
LAUNCHER: ft_launcher
Original file line number Diff line number Diff line change
Expand Up @@ -53,3 +53,4 @@ MODEL_ARGS:
--async-save: true
--use-persistent-ckpt-worker: true
TEST_TYPE: ckpt-resume
LAUNCHER: ft_launcher
Original file line number Diff line number Diff line change
Expand Up @@ -53,3 +53,4 @@ MODEL_ARGS:
--async-save: true
--use-persistent-ckpt-worker: true
TEST_TYPE: ckpt-resume
LAUNCHER: ft_launcher
Original file line number Diff line number Diff line change
Expand Up @@ -55,3 +55,4 @@ MODEL_ARGS:
--async-save: true
--use-persistent-ckpt-worker: true
TEST_TYPE: ckpt-resume
LAUNCHER: ft_launcher
Original file line number Diff line number Diff line change
Expand Up @@ -55,3 +55,4 @@ MODEL_ARGS:
--async-save: true
--use-persistent-ckpt-worker: true
TEST_TYPE: ckpt-resume
LAUNCHER: ft_launcher
Original file line number Diff line number Diff line change
Expand Up @@ -54,3 +54,4 @@ MODEL_ARGS:
--async-save: true
--use-persistent-ckpt-worker: true
TEST_TYPE: regular
LAUNCHER: ft_launcher
Original file line number Diff line number Diff line change
Expand Up @@ -56,3 +56,4 @@ MODEL_ARGS:
--async-save: true
--use-persistent-ckpt-worker: true
TEST_TYPE: regular
LAUNCHER: ft_launcher
Original file line number Diff line number Diff line change
Expand Up @@ -53,3 +53,4 @@ MODEL_ARGS:
--log-memory-to-tensorboard: true
--use-persistent-ckpt-worker: true
TEST_TYPE: regular
LAUNCHER: ft_launcher
Original file line number Diff line number Diff line change
Expand Up @@ -55,3 +55,4 @@ MODEL_ARGS:
--log-memory-to-tensorboard: true
--use-persistent-ckpt-worker: true
TEST_TYPE: frozen-resume
LAUNCHER: ft_launcher
Original file line number Diff line number Diff line change
Expand Up @@ -52,3 +52,4 @@ MODEL_ARGS:
--attention-backend: unfused
--log-memory-to-tensorboard: true
TEST_TYPE: regular
LAUNCHER: ft_launcher
Original file line number Diff line number Diff line change
Expand Up @@ -53,3 +53,4 @@ MODEL_ARGS:
--log-memory-to-tensorboard: true
--use-persistent-ckpt-worker: true
TEST_TYPE: ckpt-resume
LAUNCHER: ft_launcher
Original file line number Diff line number Diff line change
Expand Up @@ -53,3 +53,4 @@ MODEL_ARGS:
--log-memory-to-tensorboard: true
--use-persistent-ckpt-worker: true
TEST_TYPE: ckpt-resume
LAUNCHER: ft_launcher
Original file line number Diff line number Diff line change
Expand Up @@ -53,3 +53,4 @@ MODEL_ARGS:
--bf16: true
--log-memory-to-tensorboard: true
TEST_TYPE: ckpt-resume
LAUNCHER: ft_launcher
Original file line number Diff line number Diff line change
Expand Up @@ -53,3 +53,4 @@ MODEL_ARGS:
--bf16: true
--log-memory-to-tensorboard: true
TEST_TYPE: ckpt-resume
LAUNCHER: ft_launcher
Original file line number Diff line number Diff line change
Expand Up @@ -54,3 +54,4 @@ MODEL_ARGS:
--async-save: true
--use-persistent-ckpt-worker: true
TEST_TYPE: ckpt-resume
LAUNCHER: ft_launcher
Original file line number Diff line number Diff line change
Expand Up @@ -54,3 +54,4 @@ MODEL_ARGS:
--log-memory-to-tensorboard: true
--use-persistent-ckpt-worker: true
TEST_TYPE: ckpt-resume
LAUNCHER: ft_launcher
Original file line number Diff line number Diff line change
Expand Up @@ -54,3 +54,4 @@ MODEL_ARGS:
--log-memory-to-tensorboard: true
--use-persistent-ckpt-worker: true
TEST_TYPE: ckpt-resume
LAUNCHER: ft_launcher
Original file line number Diff line number Diff line change
Expand Up @@ -53,3 +53,4 @@ MODEL_ARGS:
--async-save: true
--use-persistent-ckpt-worker: true
TEST_TYPE: ckpt-resume
LAUNCHER: ft_launcher
Original file line number Diff line number Diff line change
Expand Up @@ -53,3 +53,4 @@ MODEL_ARGS:
--async-save: true
--use-persistent-ckpt-worker: true
TEST_TYPE: ckpt-resume
LAUNCHER: ft_launcher
Original file line number Diff line number Diff line change
Expand Up @@ -55,3 +55,4 @@ MODEL_ARGS:
--async-save: true
--use-persistent-ckpt-worker: true
TEST_TYPE: regular
LAUNCHER: ft_launcher
Original file line number Diff line number Diff line change
Expand Up @@ -54,3 +54,4 @@ MODEL_ARGS:
--log-memory-to-tensorboard: true
--use-persistent-ckpt-worker: true
TEST_TYPE: regular
LAUNCHER: ft_launcher
Original file line number Diff line number Diff line change
Expand Up @@ -53,3 +53,4 @@ MODEL_ARGS:
--async-save: true
--use-persistent-ckpt-worker: true
TEST_TYPE: regular
LAUNCHER: ft_launcher
Original file line number Diff line number Diff line change
Expand Up @@ -55,3 +55,4 @@ MODEL_ARGS:
--async-save: true
--use-persistent-ckpt-worker: true
TEST_TYPE: regular
LAUNCHER: ft_launcher
Original file line number Diff line number Diff line change
Expand Up @@ -55,3 +55,4 @@ MODEL_ARGS:
--async-save: true
--use-persistent-ckpt-worker: true
TEST_TYPE: regular
LAUNCHER: ft_launcher
Original file line number Diff line number Diff line change
Expand Up @@ -54,3 +54,4 @@ MODEL_ARGS:
--async-save: true
--use-persistent-ckpt-worker: true
TEST_TYPE: regular
LAUNCHER: ft_launcher
Original file line number Diff line number Diff line change
Expand Up @@ -50,3 +50,4 @@ MODEL_ARGS:
--attention-backend: unfused
--log-memory-to-tensorboard: true
TEST_TYPE: regular
LAUNCHER: ft_launcher
Original file line number Diff line number Diff line change
Expand Up @@ -55,3 +55,4 @@ MODEL_ARGS:
--async-save: true
--use-persistent-ckpt-worker: true
TEST_TYPE: regular
LAUNCHER: ft_launcher
Original file line number Diff line number Diff line change
Expand Up @@ -60,3 +60,4 @@ MODEL_ARGS:
--async-save: true
--use-persistent-ckpt-worker: true
TEST_TYPE: regular
LAUNCHER: ft_launcher
Original file line number Diff line number Diff line change
Expand Up @@ -56,3 +56,4 @@ MODEL_ARGS:
--attention-backend: unfused
--log-memory-to-tensorboard: true
TEST_TYPE: regular
LAUNCHER: ft_launcher
Original file line number Diff line number Diff line change
Expand Up @@ -56,3 +56,4 @@ MODEL_ARGS:
--attention-backend: unfused
--log-memory-to-tensorboard: true
TEST_TYPE: regular
LAUNCHER: ft_launcher
Original file line number Diff line number Diff line change
Expand Up @@ -56,3 +56,4 @@ MODEL_ARGS:
--async-save: true
--use-persistent-ckpt-worker: true
TEST_TYPE: regular
LAUNCHER: ft_launcher
Original file line number Diff line number Diff line change
Expand Up @@ -52,3 +52,4 @@ MODEL_ARGS:
--bf16: true
--log-memory-to-tensorboard: true
TEST_TYPE: regular # Usually ckpt-resume, but as a WAR to #513 set to regular
LAUNCHER: ft_launcher
Original file line number Diff line number Diff line change
Expand Up @@ -52,3 +52,4 @@ MODEL_ARGS:
--bf16: true
--log-memory-to-tensorboard: true
TEST_TYPE: regular # Usually ckpt-resume, but as a WAR to #513 set to regular
LAUNCHER: ft_launcher
Original file line number Diff line number Diff line change
Expand Up @@ -55,3 +55,4 @@ MODEL_ARGS:
--async-save: true
--use-persistent-ckpt-worker: true
TEST_TYPE: regular # Usually ckpt-resume, but as a WAR to #513 set to regular
LAUNCHER: ft_launcher
Original file line number Diff line number Diff line change
Expand Up @@ -55,3 +55,4 @@ MODEL_ARGS:
--async-save: true
--use-persistent-ckpt-worker: true
TEST_TYPE: ckpt-resume
LAUNCHER: ft_launcher
Original file line number Diff line number Diff line change
Expand Up @@ -55,3 +55,4 @@ MODEL_ARGS:
--async-save: true
--use-persistent-ckpt-worker: true
TEST_TYPE: ckpt-resume
LAUNCHER: ft_launcher
Original file line number Diff line number Diff line change
Expand Up @@ -55,3 +55,4 @@ MODEL_ARGS:
--async-save: true
--use-persistent-ckpt-worker: true
TEST_TYPE: ckpt-resume
LAUNCHER: ft_launcher
Original file line number Diff line number Diff line change
Expand Up @@ -55,3 +55,4 @@ MODEL_ARGS:
--async-save: true
--use-persistent-ckpt-worker: true
TEST_TYPE: ckpt-resume
LAUNCHER: ft_launcher
Loading
Loading