ROCm / ROCm/AMDMIGraphX

Disable offloading attention like kernels to rocMLIR if there is no wmma/mfma for requested dtype

Open
#4,308 2 comments 0 reactions 1 assignee View on GitHub

@bdevorem is already working on this.

Since Sep 19, 2025.

Dominant language
C++
Stars
333
Forks
150
Avg merge
4d 19h
Merged PRs (30d)
54

Description

DOR (Definition of Ready)

Ready

Description

rocMLIR doesn't support attention like kernels. eg. GEG, Attention, CEG if there is no associated wmma/mfma instruction for that dtype on the hardware.

e.g. using Fp32 on navi3x for the attention would crash.

DOD (Definition of Done)

Attention like kernels are not being generated for unsupported dtypes.

Contributor guide

No contributing guide indexed for this repository

First steps

  1. Read the whole issue, then the project's contributing guide.
  2. Comment on the issue to say you are picking it up — it saves two people doing the same work.
  3. Fork the repository and make your change on a branch.
  4. Open a pull request that references the issue number.

Assessment

This issue has not been assessed yet.

Get new issues in your inbox

A short digest of beginner-friendly GitHub issues.