There was an error while loading. Please reload this page.
1 parent aeb59b5 commit 76c7435Copy full SHA for 76c7435
test/tpu/run_tests.sh
@@ -11,6 +11,7 @@ python3 test/spmd/test_xla_distributed_checkpoint.py
11
python3 test/spmd/test_train_spmd_linear_model.py
12
python3 test/spmd/test_xla_spmd_python_api_interaction.py
13
python3 test/spmd/test_xla_auto_sharding.py
14
+python3 test/spmd/test_fsdp_v2.py
15
XLA_EXPERIMENTAL=nonzero:masked_select:nms python3 test/ds/test_dynamic_shape_models.py -v
16
XLA_EXPERIMENTAL=nonzero:masked_select:nms python3 test/ds/test_dynamic_shapes.py -v
17
python3 test/test_autocast.py
0 commit comments