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
..
RAEDME.md	[shardformer] fix an error in readme (#3988)	2023-07-04 16:05:01 +08:00
__init__.py	[shardformer] supported fused qkv checkpoint (#4073)	2023-07-04 16:05:01 +08:00
api.py	[shardformer] supported fused qkv checkpoint (#4073)	2023-07-04 16:05:01 +08:00
comm_spec.py	[test] fixed tests failed due to dtensor change (#4082)	2023-07-04 16:05:01 +08:00
d_tensor.py	Revert "[sync] sync feature/shardformer with develop"	2023-06-09 09:41:27 +08:00
layout.py	[test] fixed tests failed due to dtensor change (#4082)	2023-07-04 16:05:01 +08:00
layout_converter.py	[test] fixed tests failed due to dtensor change (#4082)	2023-07-04 16:05:01 +08:00
misc.py	[DTensor] refactor sharding spec (#2987)	2023-03-07 11:08:11 +08:00
sharding_spec.py	Revert "[sync] sync feature/shardformer with develop"	2023-06-09 09:41:27 +08:00
utils.py	[shardformer] support module saving and loading (#4062)	2023-07-04 16:05:01 +08:00