github/ColossalAI
1
0
Fork 0
mirror of https://github.com/hpcaitech/ColossalAI.git synced 2025-06-22 13:41:43 +00:00
Code Issues Packages Projects Releases Wiki Activity
1,290 Commits 56 Branches 51 Tags 236 MiB
9c5e028a62
Commit Graph

1290 Commits

This Branch
This Branch
All Branches
Author SHA1 Message Date
oahzxl
d31e146687 code format 2022-12-10 17:34:40 +08:00
oahzxl
929445116a pass outproduct mean 2022-12-10 17:29:51 +08:00
oahzxl
979e61db92 redesign index tracer, add source and change compute 2022-12-09 17:39:02 +08:00
oahzxl
2b4ebcc278 finishi codegen on msa 2022-12-08 15:16:10 +08:00
oahzxl
6d99994a7a rename index tracer 2022-12-06 17:35:27 +08:00
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