[shardformer] add 3d plugin to ci test

hpcaitech · Sep 4, 2023 · 5447d85 · 5447d85
1 parent b01febe
commit 5447d85
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/examples/language/bert/test_ci.sh b/examples/language/bert/test_ci.sh
@@ -3,6 +3,6 @@ set -xe
 
 pip install -r requirements.txt
 
-for plugin in "torch_ddp" "torch_ddp_fp16" "gemini" "low_level_zero"; do
+for plugin in "torch_ddp" "torch_ddp_fp16" "gemini" "low_level_zero" "hybrid_parallel"; do
    torchrun --standalone --nproc_per_node 4  finetune.py --target_f1 0.86 --plugin $plugin --model_type "bert"
 done