Commit Graph

1252 Commits

Author SHA1 Message Date
oahzxl
e66a18a0bf optimise search 2022-12-16 15:06:39 +08:00
oahzxl
e83e3c6154 update memory estimate 2022-12-16 11:09:35 +08:00
oahzxl
de65e6c3e8 support output 2022-12-13 11:00:51 +08:00
oahzxl
cda3e8572a support index dupilictae and update loop 2022-12-13 10:02:26 +08:00
oahzxl
1e0fd11bc1 support check_index_duplicate 2022-12-13 10:01:30 +08:00
oahzxl
8754fa2553 change threshold 2022-12-12 18:25:47 +08:00
oahzxl
98f9728e29 code style 2022-12-12 18:15:47 +08:00
oahzxl
8511d900a8 code style 2022-12-12 17:36:17 +08:00
oahzxl
5cdfcfe1d1 code style 2022-12-12 17:29:07 +08:00
oahzxl
b7b67c32ad code style 2022-12-12 17:25:38 +08:00
oahzxl
31a2c5d09f work with outerproductmean and msa 2022-12-12 17:24:06 +08:00
oahzxl
5de9e46381 code format 2022-12-10 17:34:48 +08:00
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