github
/
ColossalAI
mirror of https://github.com/hpcaitech/ColossalAI
1
0
Fork 0
Code Issues Projects Releases Wiki Activity
ColossalAI/colossalai/auto_parallel/tensor_shard/node_handler
History
Hongxin Liu 554aa9592e
[legacy] move communication and nn to legacy and refactor logger (#4671) 
* [legacy] move communication to legacy (#4640)

* [legacy] refactor logger and clean up legacy codes (#4654)

* [legacy] make logger independent to gpc

* [legacy] make optim independent to registry

* [legacy] move test engine to legacy

* [legacy] move nn to legacy (#4656)

* [legacy] move nn to legacy

* [checkpointio] fix save hf config

* [test] remove useledd rpc pp test

* [legacy] fix nn init

* [example] skip tutorial hybriad parallel example

* [devops] test doc check

* [devops] test doc check
 		2023-09-11 16:24:28 +08:00
..
strategy [NFC] polish unary_elementwise_generator.py code style (#4267) 2023-07-26 14:12:57 +08:00
__init__.py [autoparallel] add shard option (#2696) 2023-02-15 13:48:28 +08:00
addmm_handler.py [autoparallel] support addmm in tracer and solver (#1961) 2022-11-16 14:59:18 +08:00
batch_norm_handler.py [autoparallel]integrate auto parallel feature with new tracer (#3408) 2023-04-04 17:40:45 +08:00
binary_elementwise_handler.py [autoparallel] update binary elementwise handler (#2451) 2023-01-12 09:35:10 +08:00
bmm_handler.py [autoparallel] adapt autoparallel with new analyzer (#3261) 2023-03-30 17:47:24 +08:00
conv_handler.py [autoparallel] use metainfo in handler (#2149) 2022-12-20 10:31:22 +08:00
default_reshape_handler.py [autoparallel] refactor handlers which reshape input tensors (#2615) 2023-02-08 15:02:49 +08:00
embedding_handler.py [NFC]fix typo colossalai/auto_parallel nn utils etc. (#3779) 2023-05-23 15:28:20 +08:00
getattr_handler.py [autoparallel] fix spelling error (#2270) 2023-01-03 16:13:00 +08:00
getitem_handler.py [NFC] polish colossalai/auto_parallel/tensor_shard/node_handler/getitem_handler.py code style (#2289) 2023-01-04 15:09:57 +08:00
layer_norm_handler.py [autoparallel] Patch meta information of `torch.nn.LayerNorm` (#2647) 2023-02-10 14:29:24 +08:00
linear_handler.py [NFC]fix typo colossalai/auto_parallel nn utils etc. (#3779) 2023-05-23 15:28:20 +08:00
matmul_handler.py [nfc] fix typo colossalai/ applications/ (#3831) 2023-05-25 16:19:41 +08:00
node_handler.py [test] fixed tests failed due to dtensor change (#4082) 2023-07-04 16:05:01 +08:00
normal_pooling_handler.py [autoparallel] use metainfo in handler (#2149) 2022-12-20 10:31:22 +08:00
output_handler.py [autoparallel] fix spelling error (#2270) 2023-01-03 16:13:00 +08:00
permute_handler.py [autoparallel] refactor handlers which reshape input tensors (#2615) 2023-02-08 15:02:49 +08:00
placeholder_handler.py [autoparallel] fix spelling error (#2270) 2023-01-03 16:13:00 +08:00
registry.py [legacy] move communication and nn to legacy and refactor logger (#4671) 2023-09-11 16:24:28 +08:00
softmax_handler.py [autoparallel] implement softmax handler (#2132) 2022-12-14 16:09:53 +08:00
split_handler.py [autoparallel] refactor handlers which reshape input tensors (#2615) 2023-02-08 15:02:49 +08:00
sum_handler.py [autoparallel] add sum handler (#2101) 2022-12-08 17:02:54 +08:00
tensor_constructor_handler.py [autoparallel] add tensor constructor handler (#2082) 2022-12-06 10:20:10 +08:00
transpose_handler.py [autoparallel] refactor handlers which reshape input tensors (#2615) 2023-02-08 15:02:49 +08:00
unary_elementwise_handler.py [autockpt] make it work. (#2257) 2023-01-02 23:37:45 +08:00
view_handler.py [autoparallel] refactor handlers which reshape input tensors (#2615) 2023-02-08 15:02:49 +08:00
where_handler.py [autoparallel] adapt handlers with attention block (#1990) 2022-11-21 10:44:11 +08:00