You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
ColossalAI/colossalai/auto_parallel/tensor_shard/utils
YuliangLiu0306 1c1fe44305
[autoparallel] adapt solver with self attention (#2037)
2 years ago
..
__init__.py [autoparallel] add experimental view handler (#2011) 2 years ago
broadcast.py [autoparallel]add essential CommActions for broadcast oprands (#1793) 2 years ago
factory.py [autoparallel] refactored the autoparallel module for organization (#1706) 2 years ago
misc.py [autoparallel] use pytree map style to process data (#1989) 2 years ago
reshape.py [autoparallel] adapt solver with self attention (#2037) 2 years ago
sharding.py [autoparallel] fixed wrong sharding strategy in conv handler (#1747) 2 years ago