duanjunwen fa3ccda8ee [fix] fix send recv signature;

2024-10-29 03:33:58 +00:00

..

schedule

[fix] fix send recv signature;

2024-10-29 03:33:58 +00:00

__init__.py

[feat] add zerobubble pp (just a frame now); add POC test for dx_dw; add test for zerobubble;

2024-08-22 10:25:34 +00:00

p2p.py

fix object_to_tensor usage when torch>=2.3.0 (#5820)

2024-07-16 13:59:25 +08:00

stage_manager.py

[fix\ fix fail case test_shard_llama

2024-10-25 02:28:55 +00:00

weight_grad_store.py

[fix] fix handle name; rm useless comments;

2024-10-29 03:24:15 +00:00