Skip to content

Commit 76c7435

Browse files
committed
Add fsdpv2 test to tpu ci
1 parent aeb59b5 commit 76c7435

File tree

1 file changed

+1
-0
lines changed

1 file changed

+1
-0
lines changed

test/tpu/run_tests.sh

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -11,6 +11,7 @@ python3 test/spmd/test_xla_distributed_checkpoint.py
1111
python3 test/spmd/test_train_spmd_linear_model.py
1212
python3 test/spmd/test_xla_spmd_python_api_interaction.py
1313
python3 test/spmd/test_xla_auto_sharding.py
14+
python3 test/spmd/test_fsdp_v2.py
1415
XLA_EXPERIMENTAL=nonzero:masked_select:nms python3 test/ds/test_dynamic_shape_models.py -v
1516
XLA_EXPERIMENTAL=nonzero:masked_select:nms python3 test/ds/test_dynamic_shapes.py -v
1617
python3 test/test_autocast.py

0 commit comments

Comments
 (0)