ColossalAI/colossalai/auto_parallel/tensor_shard/node_handler
YuliangLiu0306 ea0f6b8df9
[autoparallel] add runtime pass and numerical test for view handler (#2018)
2022-11-25 15:50:16 +08:00
..
experimental [autoparallel] add runtime pass and numerical test for view handler (#2018) 2022-11-25 15:50:16 +08:00
strategy [autoparallel] use pytree map style to process data (#1989) 2022-11-21 10:44:22 +08:00
__init__.py
addmm_handler.py
batch_norm_handler.py
binary_elementwise_handler.py
bmm_handler.py
conv_handler.py
getatrr_handler.py
getitem_handler.py
layer_norm_handler.py
linear_handler.py [autoparallel] add runtime pass and numerical test for view handler (#2018) 2022-11-25 15:50:16 +08:00
matmul_handler.py
node_handler.py [autoparallel] use pytree map style to process data (#1989) 2022-11-21 10:44:22 +08:00
normal_pooling_handler.py
output_handler.py
placeholder_handler.py
registry.py
reshape_handler.py [autoparallel] adapt handlers with attention block (#1990) 2022-11-21 10:44:11 +08:00
unary_elementwise_handler.py [autoparallel] adapt handlers with attention block (#1990) 2022-11-21 10:44:11 +08:00
where_handler.py [autoparallel] adapt handlers with attention block (#1990) 2022-11-21 10:44:11 +08:00