From e710ebd7129b9d0d48b01c30d18a9606d6da6034 Mon Sep 17 00:00:00 2001 From: IvanKobzarev Date: Thu, 5 Feb 2026 12:17:30 -0800 Subject: [PATCH] [ci] Add DSv3 simple_fsdp auto_bucketing to CI stack-info: PR: https://github.com/meta-pytorch/autoparallel/pull/315, branch: IvanKobzarev/stack/24 --- .github/workflows/test_torchtitan.yml | 11 +++++++++++ 1 file changed, 11 insertions(+) diff --git a/.github/workflows/test_torchtitan.yml b/.github/workflows/test_torchtitan.yml index 9d3c3f9d..ea0adc48 100644 --- a/.github/workflows/test_torchtitan.yml +++ b/.github/workflows/test_torchtitan.yml @@ -59,3 +59,14 @@ jobs: --parallelism.data_parallel_shard_degree 4 \ --parallelism.expert_parallel_degree 4 \ --job.custom_config_module=torchtitan.experiments.autoparallel.job_config + + # TorchTitan DSv3 SimpleFSDP with auto_bucketing passes + NGPU=4 CONFIG_FILE=./torchtitan/models/deepseek_v3/train_configs/debug_model.toml ./run_train.sh \ + --model.name simple_fsdp.deepseek_v3 \ + --parallelism.tensor_parallel_degree 1 \ + --parallelism.expert_parallel_degree 4 \ + --job.custom_config_module=torchtitan.experiments.simple_fsdp.job_config \ + --compile.graph_passes auto_bucketing \ + --activation_checkpoint.mode "none" \ + --compile.backend inductor \ + --compile.enable