ColossalAI/colossalai/auto_parallel/tensor_shard/node_handler
YuliangLiu0306 49216d7ab1
[autoparallel] fix bugs caused by negative dim key (#1808)
* [autoparallel] fix bugs caused by negative dim key

* fix import error

* fix matmul test issue

* fix unit test issue
2022-11-08 17:03:50 +08:00
..
strategy [autoparallel] fix bugs caused by negative dim key (#1808) 2022-11-08 17:03:50 +08:00
__init__.py
batch_norm_handler.py
binary_elementwise_handler.py [autoparallel]add essential CommActions for broadcast oprands (#1793) 2022-11-04 18:36:42 +08:00
bmm_handler.py [autoparallel]add essential CommActions for broadcast oprands (#1793) 2022-11-04 18:36:42 +08:00
conv_handler.py
getatrr_handler.py
getitem_handler.py
layer_norm_handler.py
linear_handler.py
matmul_handler.py [autoparallel] fix bugs caused by negative dim key (#1808) 2022-11-08 17:03:50 +08:00
node_handler.py
normal_pooling_handler.py
output_handler.py
placeholder_handler.py
registry.py
reshape_handler.py [autoparallel] fix bias addition module (#1800) 2022-11-08 16:21:25 +08:00
unary_elementwise_handler.py
where_handler.py [autoparallel]add essential CommActions for broadcast oprands (#1793) 2022-11-04 18:36:42 +08:00