github/ColossalAI
1
0
Fork 0
mirror of https://github.com/hpcaitech/ColossalAI.git synced 2026-07-12 19:06:17 +00:00
Code Issues Packages Projects Releases Wiki Activity
Files
36ab2cb783fdee9899f734f99633fc0d63d4e980
ColossalAI/colossalai/auto_parallel/tensor_shard/node_handler/experimental
History
YuliangLiu0306 0dbcd4a6f5 [autoparallel] add split handler (#2032) 
* [autoparallel] add split handler

* add numerical test and runtime passes
2022-11-29 11:03:51 +08:00
..
__init__.py
[autoparallel] add split handler (#2032)
2022-11-29 11:03:51 +08:00
permute_handler.py
[autoparallel] add experimental permute handler (#2029)
2022-11-27 20:26:52 +08:00
reshape_generator.py
[autoparallel] add split handler (#2032)
2022-11-29 11:03:51 +08:00
split_handler.py
[autoparallel] add split handler (#2032)
2022-11-29 11:03:51 +08:00
transpose_handler.py
[autoparallel] add experimental permute handler (#2029)
2022-11-27 20:26:52 +08:00
view_handler.py
[autoparallel] add experimental permute handler (#2029)
2022-11-27 20:26:52 +08:00