Kernels
flash-attention-1-triton / torch-ext /torch_binding.cpp

Commit History

feat: implement Flash Attention 1 using Triton with PyTorch C++ bindings and test suite
8cf888e

sigmoid-neuron commited on