duanjunwen ab643c9af7 [fix] rm output.data after send fwd;

2024-09-03 14:12:17 +08:00

..

schedule

[fix] rm output.data after send fwd;

2024-09-03 14:12:17 +08: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

[Feature] optimize PP overlap (#5735)

2024-06-26 14:48:02 +08:00