8 Commits (0d3592a53fb5710e208089b0f7b822e1d5bf7167)

Author	SHA1	Message	Date
huangting4201	5d8313693b	feat(model/overlap_handler.py): fix head post backward hook when activation	2023-10-24 17:29:09 +08:00
yingtongxiong	97dcefc389	support model activation checkpoint	2023-10-24 16:13:52 +08:00
huangting4201	03cc7f9b80	feat(model/overlap_handler.py): fix lint error	2023-10-23 15:28:34 +08:00
huangting4201	0d693cf3a1	feat(model/overlap_handler.py): fix lint error	2023-10-23 15:22:03 +08:00
yingtongxiong	f6a5086fe4	support bias	2023-10-23 14:51:27 +08:00
huangting4201	e7f9f1d208	feat(model/overlap_handler.py): optimize reduce scatter mem pool	2023-10-23 13:31:23 +08:00
huangting4201	b20f47a1fe	feat(model/overlap_handler.py): move handler to gpc	2023-10-23 12:02:32 +08:00
huangting4201	85ad917ae4	feat(model/overlap_handler.py): refactor overlap hook handle	2023-10-20 21:50:32 +08:00