Suggestion Description
Refactor the MI355X FMHA kernel to make the implementation more modular, remove redundant code, and reduce the overall line count.
The refactor should use official FlyDSL APIs wherever possible, align the code with FlyDSL conventions, and make the kernel easier to maintain, extend, and tune.
Scope
- Split reusable MI355X FMHA logic into clearer modules.
- Remove duplicated logic and reduce the total code size.
- Prefer official FlyDSL APIs over non-standard or temporary implementations.
- Preserve the existing functionality and performance baseline while avoiding behavioral regressions.
Suggestion Description
Refactor the MI355X FMHA kernel to make the implementation more modular, remove redundant code, and reduce the overall line count.
The refactor should use official FlyDSL APIs wherever possible, align the code with FlyDSL conventions, and make the kernel easier to maintain, extend, and tune.
Scope