ROCm/composable_kernel
1
0
Fork 0
mirror of https://github.com/ROCm/composable_kernel.git synced 2026-06-28 18:56:59 +00:00
Code Issues Packages Projects Releases Wiki Activity
Files
0fdbf8a91d10df00a0635c3f2bd3959de0a126de
composable_kernel/example/ck_tile/42_mx_gemm
History
Enrico Degregori bdd7a8333d [rocm-libraries] ROCm/rocm-libraries#6672 (commit bda3f97) 
[CK Tile] PermuteN support MX GEMM

## Motivation

Add PermuteN support to preshuffle MX GEMM

## Technical Details

 - Modify `shuffle_b_permuteN` to support MX preshuffled layout
- Add `preShuffleScalePermuteN` with same functionality of
`preShuffleScale` but layout consistent with PermuteN
 - Include MX pre-processing functions in the library

## Test Plan

Add test configuration for permuteN with preshuffle (both FP4 and FP8)

## Submission Checklist

- [x] Look over the contributing guidelines at
https://github.com/ROCm/ROCm/blob/develop/CONTRIBUTING.md#pull-requests.

Co-authored-by: Cong Ma <congma13@amd.com>
2026-06-05 03:04:43 +00:00
..
CMakeLists.txt
[rocm-libraries] ROCm/rocm-libraries#5552 (commit 369c7a2)
2026-05-19 11:53:19 -07:00
mx_gemm_instance.hpp
[rocm-libraries] ROCm/rocm-libraries#6672 (commit bda3f97)
2026-06-05 03:04:43 +00:00
mx_gemm.cpp
[rocm-libraries] ROCm/rocm-libraries#6672 (commit bda3f97)
2026-06-05 03:04:43 +00:00
mx_gemm.hpp
[rocm-libraries] ROCm/rocm-libraries#6672 (commit bda3f97)
2026-06-05 03:04:43 +00:00
run_mx_gemm.inc
[rocm-libraries] ROCm/rocm-libraries#6672 (commit bda3f97)
2026-06-05 03:04:43 +00:00