Skip to main content

Mojo module

mma_rdna

RDNA-specific MMA wrapper for Wave32 WMMA operations.

This module provides MMA operations optimized for RDNA GPUs with:

  • 16x16x16 WMMA shape
  • 16-element A/B fragments
  • 8-element C/D fragments
  • Wave32 execution model

Functions

  • mma_rdna: RDNA-specific MMA operation for Wave32 WMMA.

Was this page helpful?