18 Commits (7a3dfd0c645fba51a02eb3c6ac88b4f09160ea7d)

Author	SHA1	Message	Date
Frank Lee	c4b1b65931	[test] fixed tests failed due to dtensor change (#4082) ... * [test] fixed tests failed due to dtensor change * polish code	2023-07-04 16:05:01 +08:00
digger yu	e2d81eba0d	[nfc] fix typo colossalai/ applications/ (#3831) ... * fix typo colossalai/autochunk auto_parallel amp * fix typo colossalai/auto_parallel nn utils etc. * fix typo colossalai/auto_parallel autochunk fx/passes etc. * fix typo docs/ * change placememt_policy to placement_policy in docs/ and examples/ * fix typo colossalai/ applications/	2023-05-25 16:19:41 +08:00
digger yu	9265f2d4d7	[NFC]fix typo colossalai/auto_parallel nn utils etc. (#3779) ... * fix typo colossalai/autochunk auto_parallel amp * fix typo colossalai/auto_parallel nn utils etc.	2023-05-23 15:28:20 +08:00
YuliangLiu0306	0f392d7403	[autoparallel] find repeat blocks (#2854) ... * [autoparallel] find repeat blocks * polish * polish * polish	2023-02-23 17:28:19 +08:00
binmakeswell	a881d6d000	Revert "[NFC] polish code format" (#2372)	2023-01-06 16:01:09 +08:00
LuGY	e11a005c02	[NFC] polish colossalai/auto_parallel/tensor_shard/utils/factory.py code style (#2349)	2023-01-05 21:17:42 +08:00
YuliangLiu0306	1c1fe44305	[autoparallel] adapt solver with self attention (#2037) ... * [autoparallel] adapt solver with self attention * polish code	2022-12-01 17:53:15 +08:00
YuliangLiu0306	1438993113	[autoparallel] add experimental view handler (#2011) ... * [autoparallel] add experimental view handler * polish * polish * polish code * rename variables	2022-11-24 11:34:41 +08:00
YuliangLiu0306	155891113e	[autoparallel] use pytree map style to process data (#1989)	2022-11-21 10:44:22 +08:00
YuliangLiu0306	e34e850a4c	[autoparallel]add essential CommActions for broadcast oprands (#1793)	2022-11-04 18:36:42 +08:00
Frank Lee	f3f19a5c47	[autoparallel] added matmul handler (#1763) ... * [autoparallel] added matmul handler * polish code	2022-11-01 15:14:53 +08:00
Frank Lee	f9a613d660	[autoparallel] added binary elementwise node handler (#1758) ... * [autoparallel] added binary elementwise node handler * polish code	2022-10-25 14:32:01 +08:00
Frank Lee	262652c8bc	[autoparallel] added addbmm handler (#1751)	2022-10-21 18:55:48 +08:00
Frank Lee	474111ecb5	[autoparallel] fixed wrong sharding strategy in conv handler (#1747) ... * [autoparallel] fixed wrong sharding strategy in conv handler * polish code	2022-10-20 16:12:39 +08:00
Frank Lee	8b8937d901	[autoparallel] fixed wrong generated strategy for dot op (#1746) ... * [autoparallel] fixed wrong generated strategy for dot op * polish code	2022-10-20 15:18:16 +08:00
Frank Lee	88a79814fb	[autoparallel] handled illegal strategy in node handler (#1743) ... * [autoparallel] handled illegal strategy in node handler * polish code	2022-10-19 17:08:52 +08:00
Frank Lee	eee84908d4	[autoparallel] handled illegal sharding strategy (#1728) ... * [autoparallel] handled illegal sharding strategy * polish code	2022-10-19 12:53:06 +08:00
Frank Lee	6c331a5a09	[autoparallel] refactored the autoparallel module for organization (#1706) ... * [autoparallel] refactored the autoparallel module for organization * polish code	2022-10-14 13:27:00 +08:00