Commit Graph

4 Commits (da4f7b855f0074b374bbd26837c036f2cdfa9564)

Author SHA1 Message Date
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