ColossalAI

Author	SHA1	Message	Date
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	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	7ab2db206f	adapt new fx	2023-01-10 11:56:00 +08:00
oahzxl	fd818cf144	change imports	2023-01-10 11:10:45 +08:00
oahzxl	a591d45b29	add available	2023-01-10 10:56:39 +08:00
oahzxl	7d4abaa525	add doc	2023-01-10 09:59:47 +08:00
oahzxl	0ea903b94e	rename trace_index to trace_indice	2023-01-09 17:25:13 +08:00
oahzxl	1951f7fa87	code style	2023-01-09 16:30:16 +08:00
oahzxl	212b5b1b5f	add comments	2023-01-09 16:29:33 +08:00
oahzxl	19cc64b1d3	remove autochunk_available	2023-01-09 16:06:58 +08:00
oahzxl	8a989a0d89	code style	2023-01-06 17:55:22 +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
oahzxl	c3d72f7db9	seperate reorder	2023-01-06 16:53:01 +08:00
oahzxl	2bde9d2b7f	code format	2023-01-06 14:21:49 +08:00
oahzxl	8a634af2f5	close mem and code print	2023-01-06 14:19:45 +08:00
oahzxl	1a6d2a740b	take apart chunk code gen	2023-01-06 14:14:45 +08:00