3 Commits (1168b50e3387e9440af6cbbd1e085696ad4e5831)

Author	SHA1	Message	Date
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