github
/
ColossalAI
mirror of https://github.com/hpcaitech/ColossalAI
1
0
Fork 0
Code Issues Projects Releases Wiki Activity
1,235 Commits
33 Branches
47 Tags
230 MiB
Commit Graph

1235 Commits (a9d64377bb237f34fdafaeec2abcfdfb6e080091)

Author SHA1 Message Date
oahzxl a9d64377bb support new op 2022-12-06 17:34:24 +08:00
oahzxl f24c418bb0 finish chunk define 2022-12-06 16:29:07 +08:00
oahzxl 3b7d671206 finish region search loop 2022-12-06 11:08:39 +08:00
oahzxl 7330d90745 add possible region search 2022-12-04 17:05:28 +08:00
oahzxl d9ca2f898d polish code 2022-11-15 15:50:50 +08:00
oahzxl 54a34a7e46 update active log 2022-11-15 11:30:43 +08:00
oahzxl fad3b6d1a6 polish code 2022-11-15 10:46:51 +08:00
oahzxl 7e2bd1e428 polish code 2022-11-15 10:36:02 +08:00
oahzxl f379d1a94d add doc str 2022-11-15 10:18:00 +08:00
oahzxl 70a98b8f56 add doc string 2022-11-14 23:49:48 +08:00
oahzxl c36dba07de finish basic index tracer 2022-11-14 23:38:05 +08:00
oahzxl 1607d04e81 add part of index tracer 2022-11-14 16:02:47 +08:00
oahzxl d7634af5c0 finish memory estimation 2022-11-11 15:43:03 +08:00
oahzxl 22f9c60b6b fix bug 2022-11-09 17:50:39 +08:00
oahzxl 8cca684c56 finish memory estimation 2022-11-08 14:41:57 +08:00
oahzxl 12301dd2e9 finish basic inference memory estimation 2022-11-08 10:34:14 +08:00
oahzxl d95cfe2622 basic memory 2022-11-07 18:26:13 +08:00
oahzxl c35718e8db basic chunk 2022-11-04 11:18:09 +08:00
oahzxl f8aeecef46 add meta 2022-11-03 14:33:35 +08:00
oahzxl 820ea4d056 align evoformer 2022-11-02 15:49:25 +08:00
oahzxl 86f2a31474 add evoformer 2022-11-02 15:12:08 +08:00
oahzxl 78cfe4362b basic chunk 2022-11-02 13:59:48 +08:00
oahzxl 87cddf7e14 rename and remove useless func 2022-10-27 16:40:19 +08:00
oahzxl f5c5d4c4f3 init 2022-10-27 10:57:07 +08:00
YuliangLiu0306 d2fc067231
[autoparallel] fix param hook issue in transform pass (#1755) 2022-10-24 13:13:38 +08:00
Frank Lee 262652c8bc
[autoparallel] added addbmm handler (#1751) 2022-10-21 18:55:48 +08:00
YuliangLiu0306 980ed21723
[autoparallel] shard param and buffer as expected (#1753) 
* [autoparallel] shard param and buffer as expected

* fix unit test issue
 2022-10-21 15:45:13 +08:00
YuliangLiu0306 cdb7d5e7d2
[hotfix] autoparallel unit test (#1752) 2022-10-20 19:51:38 +08:00
YuliangLiu0306 a4ce180e85
[autoparallel] add sequential order to communication actions (#1735) 2022-10-20 18:48:18 +08:00
Super Daniel b893342f95
[fx] test tracer on diffuser modules. (#1750) 
* [fx] test tracer on diffuser modules.

* [fx] shorter seq_len.

* Update requirements-test.txt
 2022-10-20 18:25:05 +08:00
Frank Lee b80b6eaa88
[autoparallel] recovered skipped test cases (#1748) 2022-10-20 16:37:33 +08:00
Frank Lee 474111ecb5
[autoparallel] fixed wrong sharding strategy in conv handler (#1747) 
* [autoparallel] fixed wrong sharding strategy in conv handler

* polish code
 2022-10-20 16:12:39 +08:00
Frank Lee 8b8937d901
[autoparallel] fixed wrong generated strategy for dot op (#1746) 
* [autoparallel] fixed wrong generated strategy for dot op

* polish code
 2022-10-20 15:18:16 +08:00
Frank Lee 993b8875b6
[autoparallel] handled illegal sharding strategy in shape consistency (#1744) 
* [autoparallel] handled illegal sharding strategy in shape consistency

* polish code
 2022-10-20 12:06:25 +08:00
Frank Lee 88a79814fb
[autoparallel] handled illegal strategy in node handler (#1743) 
* [autoparallel] handled illegal strategy in node handler

* polish code
 2022-10-19 17:08:52 +08:00
Super Daniel 30874f1692
[fx/profiler] debug the fx.profiler / add an example test script for fx.profiler (#1730) 
* [fx/profiler] add test.

* [fx] fix file names.

* [fx] add docstring and comment.

* [fx] polish profiler.py.

* [fx] fix import errors.

* [fx] fix profiler.

* [fx] fix names.
 2022-10-19 14:24:51 +08:00
Frank Lee eee84908d4
[autoparallel] handled illegal sharding strategy (#1728) 
* [autoparallel] handled illegal sharding strategy

* polish code
 2022-10-19 12:53:06 +08:00
Ziheng Qin cbe9a4cb45 [NFC] polish tests/test_layers/test_3d/test_3d.py code style (#1740) 2022-10-19 12:20:51 +08:00
lucasliunju 912eb58ea0 [NFC] polish tests/test_layers/test_3d/checks_3d/common.py code style (#1733) 2022-10-19 12:20:51 +08:00
Sze-qq 23703c9dd6 [NFC] polish colossalai/nn/metric/_utils.py code style (#1727) 2022-10-19 12:20:51 +08:00
Xue Fuzhao 754aa7c81f [NFC] polish tests/test_layers/test_3d/checks_3d/check_layer_3d.py code style (#1731) 2022-10-19 12:20:51 +08:00
xyupeng ff373a11eb [NFC] polish tests/test_layers/test_sequence/checks_seq/check_layer_seq.py code style (#1723) 2022-10-19 12:20:51 +08:00
Ofey Chan 7e62af28a0 [NFC] polish accuracy_2d.py code style (#1719) 2022-10-19 12:20:51 +08:00
Arsmart1 8860d37846 [NFC] polish .github/workflows/scripts/build_colossalai_wheel.py code style (#1721) 2022-10-19 12:20:51 +08:00
LuGY 730f88f8e1 [NFC] polish _checkpoint_hook.py code style (#1722) 2022-10-19 12:20:51 +08:00
Kai Wang (Victor Kai) b38efe4e8a [NFC] polish test_2p5d/checks_2p5d/check_operation_2p5d.py code style (#1718) 2022-10-19 12:20:51 +08:00
CsRic ea961d8fd1 [NFC] polish colossalai/zero/sharded_param/__init__.py code style (#1717) 
Co-authored-by: ric <mkkt_bkkt@mail.ustc.edu.cn>
 2022-10-19 12:20:51 +08:00
yuxuan-lou 2b49ca80a3 [NFC] polish colossalai/nn/lr_scheduler/linear.py code style (#1716) 2022-10-19 12:20:51 +08:00
binmakeswell f6389d0813 [NFC] polish tests/test_layers/test_2d/checks_2d/check_operation_2d.py code style (#1715) 2022-10-19 12:20:51 +08:00
shenggan e1d780030d [NFC] polish colossalai/nn/metric/accuracy_2p5d.py code style (#1714) 2022-10-19 12:20:51 +08:00