MaxView

Timing Test Reference Updates

Post-Training Test Results (NNX)

Environment

Log
Branch test/pipeline-scan-nnx
Commit e27fc1e97
Date 2026-04-21T01:24Z
Run ID test_pipeline_scan_nnx_20260421_005336
NNX flags pure_nnx=True enable_nnx=True pure_nnx_decoder=True
Hardware V6e-32 TPU (32 devices/slice × 1 slice)
Docker image gcr.io/cienet-cmcs/wanglance-xpk-runner-pt:test_pipeline_scan_nnx

Summary

0/7 passed.

Test Result Log
01_sft_smoke FAIL log
02_sft_nnx_ckpt FAIL log
03_sft_linen_ckpt FAIL log
07_distill_smoke FAIL log
04_rl_grpo_smoke FAIL log
05_rl_gspo_smoke FAIL log
06_rl_grpo_functional FAIL log

Output Paths

Log
Logs /usr/local/google~/maxtext/venv_runs/test_pipeline_scan_nnx_20260421_005336/nnx/logs/
GCS checkpoints gs://lance-maxtext/pt_ckpt_xpk_test_pipeline_scan_nnx_20260421_005336/
NNX seed ckpt (PT-02 NNX) gs://lance-maxtext/nnx_ckpt_feat_nnx_trainstate_and_training_loop_20260411_044231/nnx_ckpt_feat_nnx_trainstate_and_training_loop_20260411_044231/nnx_feat_nnx_trainstate_and_training_loop_20260411_044231_08_checkpoint_async_true/checkpoints/9/items
Linen seed ckpt (PT-02/03) gs://lance-maxtext/pt_seed_ckpts/pt_seed_ckpts/pt_seed_ckpt_gpt352k_linen/checkpoints/9/items
Teacher ckpt (PT-07) gs://lance-maxtext/pt_seed_ckpts/pt_seed_ckpts/pt_seed_ckpt_gpt352k_v32k_linen/checkpoints/4/items
RL ckpt (PT-06) gs://lance-maxtext/rl_ckpt_llama31_8b/rl_ckpt_llama31_8b/0/items