AMDMIGraphX icon indicating copy to clipboard operation
AMDMIGraphX copied to clipboard

Add initial integration of amdmlss mha

Open Zhaeong opened this issue 3 months ago • 0 comments

Motivation

To be able to use amdmlss library selected kernels

Technical Details

Adds mlss_mha operator, fuse_mlss pass, and mlss jit compiler files as initial proposal of instruction replacement to support mlss_mha kernels from amdmlss. Bypassing the current MHA operator decomposition into separate instructions that uses mlir kernels.

A first step at integrating amdmlss kernels into MIGraphX

Zhaeong avatar Sep 18 '25 00:09 Zhaeong