5 Commits (c552cefa93ac9f2ea95c0914f6ad485439fbf9c7)

Author	SHA1	Message	Date
flybird1111	25c57b9fb4	[fix] coloattention support flash attention 2 (#4347) ... Improved ColoAttention interface to support flash attention 2. Solved #4322	2023-08-04 13:46:22 +08:00
Frank Lee	dd14783f75	[kernel] fixed repeated loading of kernels (#2549) ... * [kernel] fixed repeated loading of kernels * polish code * polish code	2023-02-03 09:47:13 +08:00
zbian	6877121377	updated flash attention api	2022-11-15 15:25:39 +08:00
ver217	f68eddfb3d	refactor kernel (#142)	2022-01-13 16:47:17 +08:00
shenggan	5c3843dc98	add colossalai kernel module (#55)	2021-12-21 12:19:52 +08:00