youngkingdom/cutlass
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity
Files
e9a75581fe43dee610bbf948ccac51a522119f63
cutlass/examples/68_hopper_flash_mla/README.md
Junkai-Wu 15f5468872 Migrate FlashMLA codes to example. (#2135)
2025-02-26 01:29:07 -05:00

317 B
Raw Blame History

Hopper FlashMLA - Examples

The codes in this example are migrated from FlashMLA, it implements an efficient MLA decoding kernel for Hopper GPU.

Run the example

Install

python setup.py install

Run the test

python tests/test_flash_mla.py