github/ColossalAI
1
0
Fork 0
mirror of https://github.com/hpcaitech/ColossalAI.git synced 2025-10-07 18:15:56 +00:00
Code Issues Packages Projects Releases Wiki Activity
Files
1aaeb596c63752071cbbaa2477a7f2406901b70b
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