Skip to content

GitLab

Switch branch/tag
Find file Normal viewHistoryPermalink
ci_test_list_2.txt 2.08 KB
Newer Older
Min Xu's avatar
[feat] experimental MEVO layer (#840)  
Min Xu committed
1 
tests/nn/data_parallel/test_fsdp_shared_weights_mevo.py
Min Xu's avatar
[feat] [FSDP]: add experimental support to shared weights (#836)  
Min Xu committed
2 
tests/nn/data_parallel/test_fsdp_shared_weights.py
Min Xu's avatar
[fix]: Fixes an issue with pre_backward hook registering (#833)  
Min Xu committed
3 
tests/nn/data_parallel/test_fsdp_pre_backward_hook.py
Min Xu's avatar
[fix] FSDP forward pass overlap between compute and all-gather (#671)  
Min Xu committed
4 
tests/nn/data_parallel/test_fsdp_overlap.py
Min Xu's avatar
[fix]: support pytorch SyncBatchNorm under AMP & checkpointing with FSDP (#659)  
Min Xu committed
5
6
7 
tests/nn/data_parallel/test_fsdp_multiple_forward.py
tests/nn/data_parallel/test_fsdp_apply.py
tests/nn/data_parallel/test_fsdp_state_dict.py
Quentin Duval's avatar
[feat] Save FSDP metadata for offline unflattening + Consolidate checkpoints (#683)  
Quentin Duval committed
8 
tests/nn/data_parallel/test_fsdp_metadata.py
Min Xu's avatar
[chores]: make CI more efficient and update py39 env a bit (#447)  
Min Xu committed
9
10
11
12 
tests/utils/test_reduce_scatter_bucketer.py
tests/utils/test_containers.py
tests/utils/test_parallel.py
tests/utils/test_state_dict.py
Pavel Belevich's avatar
Fix pytorch version check (#716)  
Pavel Belevich committed
13 
tests/utils/test_version.py
anj-s's avatar
[chore] Rename and move checkpoint_activations from misc folder. (#654)  
anj-s committed
14
15
16
17 
tests/nn/checkpoint/test_checkpoint_activations.py
tests/nn/checkpoint/test_checkpoint_activations_norm.py
tests/nn/misc/test_grad_bucket.py
tests/nn/misc/test_param_bucket.py
Min Xu's avatar
[feat]: checkpoint and normalization (#457)  
Min Xu committed
18 
tests/nn/wrap/test_wrap.py
Min Xu's avatar
[chores]: make CI more efficient and update py39 env a bit (#447)  
Min Xu committed
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43 
tests/nn/pipe_process/test_pipe.py
tests/nn/pipe_process/test_transparency.py
tests/nn/pipe_process/test_inplace.py
tests/nn/pipe_process/test_bugs.py
tests/nn/pipe_process/conftest.py
tests/nn/pipe_process/test_rpc.py
tests/nn/model_parallel/test_initialize.py
tests/nn/model_parallel/test_random.py
tests/nn/model_parallel/test_cross_entropy.py
tests/nn/model_parallel/test_layers.py
tests/nn/pipe/test_microbatch.py
tests/nn/pipe/test_checkpoint.py
tests/nn/pipe/test_worker.py
tests/nn/pipe/test_balance.py
tests/nn/pipe/test_pipe.py
tests/nn/pipe/test_transparency.py
tests/nn/pipe/test_inplace.py
tests/nn/pipe/test_copy.py
tests/nn/pipe/test_bugs.py
tests/nn/pipe/conftest.py
tests/nn/pipe/test_pipeline.py
tests/nn/pipe/test_phony.py
tests/nn/pipe/test_deferred_batch_norm.py
tests/nn/pipe/test_dependency.py
tests/nn/pipe/test_stream.py
Min Xu's avatar
[fix] FSDP: disable single rank process group for auto_wrap_bn and fixed mixed...  
Min Xu committed
44
45 
tests/nn/moe/test_moe_layer.py
tests/nn/moe/test_top2gating.py
Min Xu's avatar
[feat] experimental MEVO layer (#840)  
Min Xu committed
46 
tests/experimental/nn/test_mevo.py
Min Xu's avatar
[fix]: support pytorch SyncBatchNorm under AMP & checkpointing with FSDP (#659)  
Min Xu committed
47
48 
tests/experimental/nn/test_multiprocess_pipe.py
tests/experimental/nn/test_sync_batchnorm.py
Min Xu's avatar
[fix] FSDP: disable single rank process group for auto_wrap_bn and fixed mixed...  
Min Xu committed
49
50 
tests/experimental/nn/ampnet_pipe_process/test_ampnet_pipe.py
tests/experimental/nn/test_offload.py
anj-s's avatar
[Offload][feature] Add auto shard functionality to remove requirement of...  
anj-s committed
51 
tests/experimental/nn/test_auto_shard.py
tmarkstrum's avatar
[feat]Adding DynamicLossScaler class for supporting optimizer updates on the CPU (#635)  
tmarkstrum committed
52 
tests/experimental/optim/test_dynamic_loss_scaler.py
Quentin Duval's avatar
[feat] layer memory tracking (#808)  
Quentin Duval committed
53 
tests/experimental/tooling/test_layer_memory_tracker.py
anj-s's avatar
[feature] Add the low level SSD APIs (#829)  
anj-s committed
54 
tests/experimental/nn/test_ssd_offload.py
anj-s's avatar
[feature]Add support for SSD offload with FSDP for eval workloads (#839)  
anj-s committed
55 
tests/nn/data_parallel/test_fsdp_offload.py