flash-attention/csrc
Sophia Wisdom 37e32febba
Remove commented out code in bwd (#512)
* Remove lots of comments

* Remove unused traits
2023-09-01 16:43:58 -07:00
..
cutlass@34fd98056b bump cutlass submodule (#504) 2023-08-30 10:32:04 -07:00
flash_attn Remove commented out code in bwd (#512) 2023-09-01 16:43:58 -07:00
ft_attention [ft_attention] Fix for seqlen=8136 (#488) 2023-08-28 10:00:22 -07:00
fused_dense_lib [FusedDense] Allocate lt_workspace on input device 2023-05-30 14:17:26 -07:00
fused_softmax Add Megatron attention implementation for benchmarking 2022-10-23 23:04:16 -07:00
layer_norm Fix random state for dropout_layer_norm (#315) 2023-07-23 15:05:13 -07:00
rotary Support H100 for other CUDA extensions 2023-03-15 16:59:27 -07:00
xentropy Support H100 for other CUDA extensions 2023-03-15 16:59:27 -07:00