diff --git a/test/prototype/moe_training/test_everything.sh b/test/prototype/moe_training/test_everything.sh index 79b5cf3c15..5270f21517 100755 --- a/test/prototype/moe_training/test_everything.sh +++ b/test/prototype/moe_training/test_everything.sh @@ -12,8 +12,9 @@ IS_ROCM=$(rocm-smi --version || true) # These tests do not work on ROCm yet if [ -z "$IS_ROCM" ] then -pytest test/prototype/moe_training/test_kernels.py -s -pytest test/prototype/moe_training/test_training.py -s +pytest test/prototype/moe_training/test_kernels.py -s -v +pytest test/prototype/moe_training/test_scaled_grouped_mm.py -s -v +pytest test/prototype/moe_training/test_training.py -s -v ./test/prototype/moe_training/test_fsdp.sh ./test/prototype/moe_training/test_tp.sh ./test/prototype/moe_training/test_fsdp_tp.sh diff --git a/test/prototype/moe_training/test_fsdp.sh b/test/prototype/moe_training/test_fsdp.sh index 5f858061f4..15b57991d4 100755 --- a/test/prototype/moe_training/test_fsdp.sh +++ b/test/prototype/moe_training/test_fsdp.sh @@ -1 +1 @@ -torchrun --nproc_per_node=2 --local-ranks-filter=0 -m pytest test/prototype/moe_training/test_fsdp.py -s +torchrun --nproc_per_node=2 --local-ranks-filter=0 -m pytest test/prototype/moe_training/test_fsdp.py -s -v diff --git a/test/prototype/moe_training/test_fsdp_tp.sh b/test/prototype/moe_training/test_fsdp_tp.sh index 4c00dcd853..76648bb65b 100755 --- a/test/prototype/moe_training/test_fsdp_tp.sh +++ b/test/prototype/moe_training/test_fsdp_tp.sh @@ -1 +1 @@ -torchrun --nproc_per_node=4 --local-ranks-filter=0 -m pytest test/prototype/moe_training/test_fsdp_tp.py -s +torchrun --nproc_per_node=4 --local-ranks-filter=0 -m pytest test/prototype/moe_training/test_fsdp_tp.py -s -v diff --git a/test/prototype/moe_training/test_tp.sh b/test/prototype/moe_training/test_tp.sh index 2ab7636113..736f0e06f4 100755 --- a/test/prototype/moe_training/test_tp.sh +++ b/test/prototype/moe_training/test_tp.sh @@ -1 +1 @@ -torchrun --nproc_per_node=2 --local-ranks-filter=0 -m pytest test/prototype/moe_training/test_tp.py -s +torchrun --nproc_per_node=2 --local-ranks-filter=0 -m pytest test/prototype/moe_training/test_tp.py -s -v