Skip to content

GitLab

Switch branch/tag
Find file Normal viewHistoryPermalink
ci_test_list_1.txt 526 Bytes
Newer Older
Min Xu's avatar
[feat] save memory by using bucket buffer only in backward (#633)  
Min Xu committed
1 
tests/nn/data_parallel/test_fsdp_memory.py
shuyingsunshine21's avatar
[FSDP] relax checking root condition (#620)  
shuyingsunshine21 committed
2 
tests/nn/data_parallel/test_fsdp_multiple_wrapping.py
Min Xu's avatar
[fix]: let FSDP handle model with multiple forward pass and checkpoint (#621)  
Min Xu committed
3 
tests/nn/data_parallel/test_fsdp_freezing_weights.py
Min Xu's avatar
[fix]: support pytorch SyncBatchNorm under AMP & checkpointing with FSDP (#659)  
Min Xu committed
4
5 
tests/nn/data_parallel/test_fsdp_regnet.py
tests/nn/data_parallel/test_fsdp_uneven.py
Myle Ott's avatar
FSDP: supporting gradient accumulation without no_sync context manager to save GPU memory (#752)  
Myle Ott committed
6 
tests/nn/data_parallel/test_fsdp_grad_acc.py
Min Xu's avatar
[fix]: support pytorch SyncBatchNorm under AMP & checkpointing with FSDP (#659)  
Min Xu committed
7
8
9 
tests/nn/data_parallel/test_fsdp_summon_full_params.py
tests/nn/data_parallel/test_fsdp_input.py
tests/nn/data_parallel/test_fsdp_optimizer_utils.py
Min Xu's avatar
[fix]: let FSDP handle model with multiple forward pass and checkpoint (#621)  
Min Xu committed
10 
tests/nn/data_parallel/test_fsdp.py
Pete's avatar
Use original forward pass directly when in eval mode from within checkpoint wrapper (#709)  
Pete committed
11 
tests/nn/data_parallel/test_fsdp_with_checkpoint_wrapper.py