14 Commits (578374d0dec613d1d1f6595c01a925168c26d4fc)

Author	SHA1	Message	Date
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	72341e65f4	[auto-chunk] support extramsa (#3) (#2504)	2023-01-20 10:13:03 +08:00
oahzxl	ecccc91f21	[autochunk] support autochunk on evoformer (#2497)	2023-01-19 11:41:00 +08:00
oahzxl	4953b4ace1	[autochunk] support evoformer tracer (#2485) ... support full evoformer tracer, which is a main module of alphafold. previously we just support a simplifed version of it. 1. support some evoformer's op in fx 2. support evoformer test 3. add repos for test code	2023-01-16 19:25:05 +08:00
oahzxl	1be0ac3cbf	add doc for trace indice	2023-01-09 17:59:52 +08:00
oahzxl	0b6af554df	remove useless function	2023-01-09 17:46:43 +08:00
oahzxl	d914a21d64	rename	2023-01-09 17:45:36 +08:00
oahzxl	865f2e0196	rename	2023-01-09 17:42:25 +08:00
oahzxl	a4ed5b0d0d	rename in doc	2023-01-09 17:41:26 +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