Frank Lee
|
80eba05b0a
|
[test] refactor tests with spawn (#3452)
* [test] added spawn decorator
* polish code
* polish code
* polish code
* polish code
* polish code
* polish code
|
2023-04-06 14:51:35 +08:00 |
|
Xuanlei Zhao
|
30dd13c450
|
[autochunk] support complete benchmark (#3121)
* refact memory code
* dont log free var memory
* add memory align
* update chunk target
* update setting for new memory
* finish test
* update tracer
* update typo
* update test
* add unet test
* add bench
* update bench
* update bench
* init
* support vit
* move to cpu
* add cpu benchmark
|
2023-03-13 17:42:37 +08:00 |
|
Xuanlei Zhao
|
2ca9728cbb
|
[autochunk] refactor chunk memory estimation (#2762)
* refact memory code
* dont log free var memory
* add memory align
* update chunk target
* update setting for new memory
* finish test
* update tracer
* update typo
* update test
|
2023-03-08 16:22:30 +08:00 |
|
oahzxl
|
c4b15661d7
|
[autochunk] add benchmark for transformer and alphafold (#2543)
|
2023-02-02 15:06:43 +08:00 |