17 Commits (7d8d825681a41df62f9f546d8586c50a3757e3a8)

Author	SHA1	Message	Date
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	6ba8364881	[autochunk] support diffusion for autochunk (#2621) ... * add alphafold benchmark * renae alphafold test * rename tests * rename diffuser * renme * rename * update transformer * update benchmark * update benchmark * update bench memory * update transformer benchmark * rename * support diffuser * support unet metainfo prop * fix bug and simplify code * update linear and support some op * optimize max region search, support conv * update unet test * support some op * support groupnorm and interpolate * update flow search * add fix dim in node flow * fix utils * rename * support diffusion * update diffuser * update chunk search * optimize imports * import * finish autochunk	2023-02-07 16:32:45 +08:00
oahzxl	05671fcb42	[autochunk] support multi outputs chunk search (#2538) ... Support multi outputs chunk search. Previously we only support single output chunk search. It is more flexible and improve performance by a large margin. For transformer, we reduce memory by 40% than previous search strategy. 1. rewrite search strategy to support multi outputs chunk search 2. fix many, many bugs 3. update tests	2023-02-01 13:18:51 +08:00
oahzxl	63199c6687	[autochunk] support transformer (#2526)	2023-01-31 16:00:06 +08:00
oahzxl	c04f183237	[autochunk] support parsing blocks (#2506)	2023-01-20 11:18:17 +08:00
oahzxl	ecccc91f21	[autochunk] support autochunk on evoformer (#2497)	2023-01-19 11:41:00 +08:00
oahzxl	615e7e68d9	update doc	2023-01-10 10:44:07 +08:00
oahzxl	865f2e0196	rename	2023-01-09 17:42:25 +08:00
oahzxl	1bb1f2ad89	rename	2023-01-09 17:38:16 +08:00
oahzxl	cb9817f75d	rename function from index to indice	2023-01-09 17:34:30 +08:00
oahzxl	0ea903b94e	rename trace_index to trace_indice	2023-01-09 17:25:13 +08:00
oahzxl	065f0b4c27	add doc for search	2023-01-09 17:11:51 +08:00
oahzxl	a68d240ed5	add doc for search chunk	2023-01-09 16:54:08 +08:00
oahzxl	4d223e18a2	fix typo	2023-01-09 13:46:17 +08:00
oahzxl	c3a2bf48b4	code style	2023-01-06 17:31:59 +08:00
oahzxl	a6cdbf9161	seperate trace flow	2023-01-06 17:24:23 +08:00
oahzxl	da4076846d	rename	2023-01-06 17:09:37 +08:00