Skip to content

[FlyDSL]: MI355 FMHA kernel refactor for modularity and code simplification #843

Description

@yanguahe

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.

Metadata

Metadata

Assignees

Labels

No labels
No labels

Type

No type

Fields

No fields configured for issues without a type.

Projects

No projects

Milestone

No milestone

Relationships

None yet

Development

No branches or pull requests

Issue actions