ColossalAI

Author	SHA1	Message	Date
YuliangLiu0306	fea3cb661c	[autoparallel] support addmm in tracer and solver (#1961 ) * [fx] patch addmm * [autoparallel] support addmm in tracer and solver	2022-11-16 14:59:18 +08:00
YuliangLiu0306	f6032ddb17	[autoparallel] fix bias addition module (#1800 )	2022-11-08 16:21:25 +08:00
YuliangLiu0306	e859380bf7	[fx] support module with bias addition (#1780 ) * [autoparallel] refactor tracer to fix bias addition issue * [fx] support module with bias addition * create bias_addition_module * refactor file structure * polish code * fix unit test	2022-11-01 22:53:51 +08:00
Frank Lee	262652c8bc	[autoparallel] added addbmm handler (#1751 )	2022-10-21 18:55:48 +08:00
YuliangLiu0306	413c053453	[autoparallel] add cost graph class (#1481 ) * [autoparallel] add cost graph class * polish code	2022-08-25 17:19:59 +08:00
Frank Lee	75abc75c15	[fx] fixed compatiblity issue with torch 1.10 (#1331 )	2022-07-18 11:41:27 +08:00
Frank Lee	7531c6271f	[fx] refactored the file structure of patched function and module (#1238 ) * [fx] refactored the file structure of patched function and module * polish code	2022-07-12 15:01:58 +08:00