duanjunwen da3220f48c [fix] fix pipeline util func deallocate --> release_tensor_data; fix bwd_b loss bwd branch;		2024-09-20 09:48:35 +00:00
..
schedule	[fix] fix pipeline util func deallocate --> release_tensor_data; fix bwd_b loss bwd branch;	2024-09-20 09:48:35 +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	[Feature] optimize PP overlap (#5735)	2024-06-26 14:48:02 +08:00