david/flash-attention

mirror of https://github.com/Dao-AILab/flash-attention

Author	SHA1 Message	Date
Tri Dao	36587c01cb [LayerNorm] Update layer_norm_linear	9 months ago
Tri Dao	bdcae547c7 [LayerNorm] Don't exit early in the backward pass (fix #781)	11 months ago
Tri Dao	c9861a032d [LayerNorm] Initialize mean and rstd tensor using x.device	1 year ago
Tri Dao	f5b308e258 [LayerNorm] Rename layernorm.py -> layer_norm.py	1 year ago