squall/flash-attention
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity
a8fec99a9a
flash-attention/csrc/xentropy
History
Tri Dao 7c9953815a Add fused cross entropy loss
2022-11-12 21:58:41 -08:00
..
interface.cpp Add fused cross entropy loss 2022-11-12 21:58:41 -08:00
setup.py Add fused cross entropy loss 2022-11-12 21:58:41 -08:00
xentropy_kernel.cu Add fused cross entropy loss 2022-11-12 21:58:41 -08:00