Accelerating PyTorch on AMD GPUs with FlyDSL
FlyDSL provides a high-performance backend for PyTorch on AMD MI350-series GPUs, offering significant speedups for dense and grouped GEMM, MXFP8/MXFP4 operations, RMSNorm, and TopK through explicit hardware mapping and efficient dataflow scheduling.