vllm/tests/kernels
Woosuk Kwon c9d5b6d4a8
Replace FlashAttention with xformers (#70)
2023-05-05 02:01:08 -07:00
..
activation.py Replace FlashAttention with xformers (#70) 2023-05-05 02:01:08 -07:00
attention.py Replace FlashAttention with xformers (#70) 2023-05-05 02:01:08 -07:00
cache.py Replace FlashAttention with xformers (#70) 2023-05-05 02:01:08 -07:00
layernorm.py Replace FlashAttention with xformers (#70) 2023-05-05 02:01:08 -07:00
pos_encoding.py Replace FlashAttention with xformers (#70) 2023-05-05 02:01:08 -07:00