ColossalAI

Author	SHA1	Message	Date
digger yu	70c8cdecf4	[nfc] fix typo colossalai/cli fx kernel (#3847 ) * fix typo colossalai/autochunk auto_parallel amp * fix typo colossalai/auto_parallel nn utils etc. * fix typo colossalai/auto_parallel autochunk fx/passes etc. * fix typo docs/ * change placememt_policy to placement_policy in docs/ and examples/ * fix typo colossalai/ applications/ * fix typo colossalai/cli fx kernel	2023-06-02 15:02:45 +08:00
zbian	7bc0afc901	updated flash attention usage	2023-03-20 17:57:04 +08:00
Frank Lee	918bc94b6b	[triton] added copyright information for flash attention (#2835 ) * [triton] added copyright information for flash attention * polish code	2023-02-21 11:25:57 +08:00
アマデウス	077a66dd81	updated attention kernel (#2133 )	2022-12-16 10:54:03 +08:00
zbian	6877121377	updated flash attention api	2022-11-15 15:25:39 +08:00
oahzxl	9639ea88fc	[kernel] more flexible flashatt interface (#1804 )	2022-11-07 17:02:09 +08:00
oahzxl	501a9e9cd2	[hotfix] polish flash attention (#1802 )	2022-11-07 14:30:22 +08:00
Jiarui Fang	c248800359	[kernel] skip tests of flash_attn and triton when they are not available (#1798 )	2022-11-07 13:41:13 +08:00
oahzxl	25952b67d7	[feat] add flash attention (#1762 )	2022-10-26 16:15:52 +08:00