github
/
ColossalAI
mirror of https://github.com/hpcaitech/ColossalAI
1
0
Fork 0
Code Issues Projects Releases Wiki Activity
ColossalAI/tests/test_booster/test_plugin
History
flybird11111 77ec773388
[zero]remove registered gradients hooks (#5687) 
* remove registered hooks

fix

fix

fix zero

fix

fix

fix

fix

fix zero

fix zero

fix

fix

fix

* fix

fix

fix
 		2024-05-07 12:01:38 +08:00
..
test_3d_plugin.py [misc] refactor launch API and tensor constructor (#5666) 2024-04-29 10:40:11 +08:00
test_dp_plugin_base.py [misc] refactor launch API and tensor constructor (#5666) 2024-04-29 10:40:11 +08:00
test_gemini_plugin.py [misc] refactor launch API and tensor constructor (#5666) 2024-04-29 10:40:11 +08:00
test_low_level_zero_plugin.py [zero]remove registered gradients hooks (#5687) 2024-05-07 12:01:38 +08:00
test_torch_ddp_plugin.py [misc] refactor launch API and tensor constructor (#5666) 2024-04-29 10:40:11 +08:00
test_torch_fsdp_plugin.py [misc] refactor launch API and tensor constructor (#5666) 2024-04-29 10:40:11 +08:00