Skip to content

Commit 39c406f

Browse files
committed
add more fsdp,fsdp2 cases
1 parent 24fab67 commit 39c406f

File tree

1 file changed

+40
-21
lines changed

1 file changed

+40
-21
lines changed

test/xpu/skip_list_dist.py

Lines changed: 40 additions & 21 deletions
Original file line numberDiff line numberDiff line change
@@ -1,35 +1,54 @@
11
skip_dict = {
22
"../../../../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,
35
"../../../../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,
77
"../../../../test/distributed/fsdp/test_fsdp_comm.py": None,
88
"../../../../test/distributed/fsdp/test_fsdp_comm_hooks.py": None,
99
"../../../../test/distributed/fsdp/test_fsdp_core.py": None,
1010
"../../../../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,
1533
"../../../../test/distributed/fsdp/test_fsdp_fx.py": None,
1634
"../../../../test/distributed/fsdp/test_fsdp_input.py": None,
1735
"../../../../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,
2750
"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,
3352
"../../../../test/distributed/_tools/test_fsdp2_mem_tracker.py": None,
3453
"../../../../test/distributed/_tools/test_mem_tracker.py": None,
3554
"../../../../test/distributed/_tools/test_memory_tracker.py": None,

0 commit comments

Comments
 (0)