83 Commits (4d6e1284cbe127fbe958e8fef1ca43038c6f079a)

Author	SHA1	Message	Date
YuliangLiu0306	c7ac0f4ab2	[autoparallel] add elementwise handler (#1622) ... * [autoparallel] add elementwise handler * polish code * polish code * reduce skipped strategies range * polish code	2 years ago
YuliangLiu0306	3a46215135	[autoparallel] add embedding handler (#1620)	2 years ago
YuliangLiu0306	69448f64c4	[autoparallel] protect bcast handler from invalid strategies (#1631)	2 years ago
YuliangLiu0306	0c703189b9	[autoparallel] add layernorm handler (#1629)	2 years ago
YuliangLiu0306	bf77d3ab65	[autoparallel] recover the merged node strategy index (#1613)	2 years ago
Frank Lee	d925122020	[autoparallel] added new linear module handler (#1616)	2 years ago
YuliangLiu0306	6a8f8cc05e	[hotfix] got sliced types (#1614)	2 years ago
Frank Lee	d397842fa8	[autoparallel] added new node handler (#1612)	2 years ago
YuliangLiu0306	7d1bb71d5d	[fx] PoC of runtime shape consistency application (#1607) ... * [fx] PoC of runtime shape consistency application * polish code	2 years ago
YuliangLiu0306	47b11c432c	[autoparallel]add bcast matmul strategies (#1605)	2 years ago
Frank Lee	edb67cb378	[autoparallel] refactored the data structure for sharding strategy (#1610)	2 years ago
YuliangLiu0306	eac1b79371	[autoparallel] add bcast op handler (#1600) ... * [autoparallel] add bcast op handler * polish code * add more BCAST FUNC OP * polish code * add exception handler * polish	2 years ago
Frank Lee	3abf98a633	[autoparallel] added all non-bcast matmul strategies (#1603)	2 years ago
Frank Lee	db98b695b2	[autoparallel] added strategy generator and bmm strategies (#1602)	2 years ago
YuliangLiu0306	faa23b9d9a	[autoparallel] add reshape handler (#1594) ... * [autoparallel] add reshape handler * polish code	2 years ago
Frank Lee	27fe8af60c	[autoparallel] refactored shape consistency to remove redundancy (#1591) ... * [autoparallel] refactored shape consistency to remove redundancy * polish code * polish code * polish code	2 years ago
YuliangLiu0306	d164449d00	[autoparallel] add resnet autoparallel unit test and add backward weight communication cost (#1589)	2 years ago
Frank Lee	7c18a588c8	[autoparallel] added generate_sharding_spec to utils (#1590)	2 years ago
Frank Lee	219f66c571	[autoparallel] added solver option dataclass (#1588)	2 years ago
YuliangLiu0306	82d4376c23	[autoparallel] adapt solver with resnet (#1583) ... * [autoparallel]adapt solver with resnet * polish code * polish code	2 years ago
YuliangLiu0306	0908d0fc61	[autoparallel]add backward cost info into strategies (#1524)	2 years ago
YuliangLiu0306	1a3599410d	[autoparallel] support fucntion in operator handler (#1529)	2 years ago
YuliangLiu0306	44c866a3e3	[autoparallel] change the merge node logic (#1533)	2 years ago
YuliangLiu0306	3345c6d352	[autoparellel]add strategies constructor (#1505) ... * [autoparellel]add strategies constructor * remove duplicated strategies * polish code * adapt cost graph with StrategiesConstructor * polish	2 years ago
Frank Lee	a0436a62ee	[autoparallel] added liveness analysis (#1516) ... * [autoparallel] added liveness analysis * remove memory cost	2 years ago
YuliangLiu0306	8b7d6bd5be	[autoparallel] add more sharding strategies to conv (#1487)	2 years ago
YuliangLiu0306	413c053453	[autoparallel] add cost graph class (#1481) ... * [autoparallel] add cost graph class * polish code	2 years ago
Frank Lee	d39e11dffb	[autoparallel] added namespace constraints (#1490)	2 years ago
Frank Lee	ede326298b	[autoparallel] integrate auto parallel with torch fx (#1479)	2 years ago
Frank Lee	628c7e3fc8	[autoparallel] added dot handler (#1475)	2 years ago
Frank Lee	9dae9bb2bc	[autoparallel] introduced baseclass for op handler and reduced code redundancy (#1471) ... * [autoparallel] introduced baseclass for op handler and reduced code redundancy * polish code	2 years ago
Frank Lee	3a54e1c9b7	[autoparallel] standardize the code structure (#1469)	2 years ago
YuliangLiu0306	26a37b5cd5	[autoparallel] Add conv handler to generate strategies and costs info for conv (#1467)	2 years ago