Tri Dao
|
65f723bb9a
|
Split bwd into more .cu files to speed up compilation
|
2024-07-23 01:32:09 -07:00 |
|
Tri Dao
|
751c762c9c
|
Don't specialize for hdim 224 to speed up compilation
|
2024-07-23 00:13:54 -07:00 |
|
Tri Dao
|
908511b2b6
|
Split into more .cu files to speed up compilation
|
2024-07-10 00:24:04 -07:00 |
|
Tri Dao
|
ea8a25ca38
|
Remove configure in bwd kernel launch
|
2024-01-21 15:28:33 -08:00 |
|
Tri Dao
|
b1fbbd8337
|
Implement splitKV attention
|
2023-08-29 00:58:29 -07:00 |
|
Tri Dao
|
7a983df742
|
Use generate_kernels.py script from Driss Guessous
|
2023-08-28 13:34:12 -07:00 |
|