|
1 | 1 | skip_dict = { |
2 | 2 | "../../../../test/distributed/fsdp/test_fsdp_checkpoint.py": None, |
| 3 | + "../../../../test/distributed/fsdp/test_checkpoint_wrapper.py": None, |
| 4 | + "../../../../test/distributed/fsdp/test_fsdp_backward_prefetch.py": None, |
3 | 5 | "../../../../test/distributed/fsdp/test_fsdp_apply.py": None, |
4 | | - "../../../../test/distributed/fsdp/test_fsdp_clip_grad_norm.py": ( |
5 | | - "test_ddp_parity_xpu", |
6 | | - ), |
| 6 | + "../../../../test/distributed/fsdp/test_fsdp_clip_grad_norm.py": None, |
7 | 7 | "../../../../test/distributed/fsdp/test_fsdp_comm.py": None, |
8 | 8 | "../../../../test/distributed/fsdp/test_fsdp_comm_hooks.py": None, |
9 | 9 | "../../../../test/distributed/fsdp/test_fsdp_core.py": None, |
10 | 10 | "../../../../test/distributed/fsdp/test_fsdp_exec_order.py": None, |
11 | | - "../../../../test/distributed/fsdp/test_fsdp_fine_tune.py": ( |
12 | | - "test_parity_with_non_frozen_fsdp_xpu", |
13 | | - "test_parity_with_ddp_xpu", |
14 | | - ), |
| 11 | + "../../../../test/distributed/fsdp/test_fsdp_dtensor_state_dict.py": None, |
| 12 | + "../../../../test/distributed/fsdp/test_fsdp_flatten_params.py": None, |
| 13 | + "../../../../test/distributed/fsdp/test_fsdp_freezing_weights.py": None, |
| 14 | + "../../../../test/distributed/fsdp/test_fsdp_hybrid_shard.py": None, |
| 15 | + "../../../../test/distributed/fsdp/test_fsdp_ignored_modules.py": None, |
| 16 | + "../../../../test/distributed/fsdp/test_fsdp_memory.py": None, |
| 17 | + "../../../../test/distributed/fsdp/test_fsdp_meta.py": None, |
| 18 | + "../../../../test/distributed/fsdp/test_fsdp_misc.py": None, |
| 19 | + "../../../../test/distributed/fsdp/test_fsdp_overlap.py": None, |
| 20 | + "../../../../test/distributed/fsdp/test_fsdp_pure_fp16.py": None, |
| 21 | + "../../../../test/distributed/fsdp/test_fsdp_sharded_grad_scaler.py": None, |
| 22 | + "../../../../test/distributed/fsdp/test_fsdp_state_dict.py": None, |
| 23 | + "../../../../test/distributed/fsdp/test_fsdp_tp_integration.py": None, |
| 24 | + "../../../../test/distributed/fsdp/test_fsdp_traversal.py": None, |
| 25 | + "../../../../test/distributed/fsdp/test_fsdp_uneven.py": None, |
| 26 | + "../../../../test/distributed/fsdp/test_fsdp_unshard_params.py": None, |
| 27 | + "../../../../test/distributed/fsdp/test_fsdp_use_orig_params.py": None, |
| 28 | + "../../../../test/distributed/fsdp/test_hsdp_dtensor_state_dict.py": None, |
| 29 | + "../../../../test/distributed/fsdp/test_shard_utils.py": None, |
| 30 | + "../../../../test/distributed/fsdp/test_utils.py": None, |
| 31 | + "../../../../test/distributed/fsdp/test_wrap.py": None, |
| 32 | + "../../../../test/distributed/fsdp/test_fsdp_fine_tune.py": None, |
15 | 33 | "../../../../test/distributed/fsdp/test_fsdp_fx.py": None, |
16 | 34 | "../../../../test/distributed/fsdp/test_fsdp_input.py": None, |
17 | 35 | "../../../../test/distributed/fsdp/test_fsdp_multiple_forward.py": None, |
18 | | - "../../../../test/distributed/fsdp/test_fsdp_multiple_wrapping.py": ( |
19 | | - "test_transformer_no_grad_mixed_precision_True_xpu", |
20 | | - ), |
21 | | - "../../../../test/distributed/fsdp/test_fsdp_uneven.py": None, |
22 | | - "../../../../test/distributed/fsdp/test_utils.py": None, |
23 | | - "distributed/test_c10d_xccl.py": ( |
24 | | - # https://github.com/intel/torch-xpu-ops/issues/2046 |
25 | | - "test_unwaited", |
26 | | - ), |
| 36 | + "../../../../test/distributed/fsdp/test_fsdp_multiple_wrapping.py": None, |
| 37 | + "../../../../test/distributed/_composable/fsdp/test_fully_shard_clip_grad_norm_.py": None, |
| 38 | + "../../../../test/distributed/_composable/fsdp/test_fully_shard_comm.py": None, |
| 39 | + "../../../../test/distributed/_composable/fsdp/test_fully_shard_compile.py": None, |
| 40 | + "../../../../test/distributed/_composable/fsdp/test_fully_shard_frozen.py": None, |
| 41 | + "../../../../test/distributed/_composable/fsdp/test_fully_shard_mixed_precision.py": None, |
| 42 | + "../../../../test/distributed/_composable/fsdp/test_fully_shard_state_dict.py": None, |
| 43 | + "../../../../test/distributed/_composable/fsdp/test_fully_shard_training.py": None, |
| 44 | + "../../../../test/distributed/_composable/test_composability/test_2d_composability.py": None, |
| 45 | + "../../../../test/distributed/_composable/test_replicate_with_compiler.py": None, |
| 46 | + "../../../../test/distributed/_composable/test_composability/test_pp_composability.py": None, |
| 47 | + "../../../../test/distributed/_composable/test_checkpoint.py": None, |
| 48 | + "../../../../test/distributed/_composable/test_contract.py": None, |
| 49 | + "distributed/test_c10d_xccl.py": None, |
27 | 50 | "distributed/test_c10d_ops_xccl.py": None, |
28 | | - "../../../../test/distributed/fsdp/test_fsdp_misc.py": None, |
29 | | - "../../../../test/distributed/test_functional_api.py": ( |
30 | | - # depends on https://github.com/pytorch/pytorch/pull/159473 |
31 | | - "test_tracing_with_fakepg_xpu", |
32 | | - ), |
| 51 | + "../../../../test/distributed/test_functional_api.py": None, |
33 | 52 | "../../../../test/distributed/_tools/test_fsdp2_mem_tracker.py": None, |
34 | 53 | "../../../../test/distributed/_tools/test_mem_tracker.py": None, |
35 | 54 | "../../../../test/distributed/_tools/test_memory_tracker.py": None, |
|
0 commit comments