FlashMLA/tests
Kevin Zhang e0557deb3a Feature:Support flashMLA decoding via flashAttn2(#29)
Changes:
1. Implement flashMLA with matrix absorption algorithm via flashAttn2
2. Add golden test on MXMACA platform
2025-02-24 23:56:05 +08:00
..
test_flash_mla.py Feature:Support flashMLA decoding via flashAttn2(#29) 2025-02-24 23:56:05 +08:00