Skip to content

Rework WMMA#892

Merged
pxl-th merged 8 commits intomasterfrom
pxl-th/wmma-tr
Mar 15, 2026
Merged

Rework WMMA#892
pxl-th merged 8 commits intomasterfrom
pxl-th/wmma-tr

Conversation

@pxl-th
Copy link
Copy Markdown
Member

@pxl-th pxl-th commented Mar 14, 2026

  • Move to WMMA module, remove WMMA_ prefixes.
  • Add RowMajor, ColMajor fragment loading options.
  • Allow loading C fragment from FP16 / BFP16 types.

Enables WMMA usage in flash attention implementation: FluxML/NNkernels.jl#29.

@pxl-th pxl-th merged commit e522164 into master Mar 15, 2026
3 checks passed
@pxl-th pxl-th deleted the pxl-th/wmma-tr branch March 15, 2026 14:25
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

1 participant