mirror of https://github.com/ROCm/composable_kernel.git synced 2026-05-15 02:27:57 +00:00

Files

Rostyslav Geyyer a281825509 Add a host mx gemm reference kernel (#1864 )

* Add mx gemm reference kernel

* Update copyright year

* Update mx gemm example

* Use element-wise ops in the reference gemm

[ROCm/composable_kernel commit: 37bfa01c0d]

2025-02-07 15:07:07 -07:00

CMakeLists.txt

Merge from internal (#1857 )

2025-02-07 15:05:05 -07:00

gemm_mx_common.hpp

Add a host mx gemm reference kernel (#1864 )

2025-02-07 15:07:07 -07:00

gemm_mx_fp8.cpp

Merge from internal (#1857 )

2025-02-07 15:05:05 -07:00

README.md

Merge from internal (#1857 )

2025-02-07 15:05:05 -07:00

README.md

GEMM Examples for Microscaling Formats

example_gemm_mx_fp8

# arg1: verification (0=no, 1=CPU)
# arg2: initialization (0=no init, 1=integer value, 2=decimal value)
# arg3: time kernel (0=no, 1=yes)
# arg4: verbosity (0=no info, 1=verbose info)
# arg5 to 10: M (16x), N(16x), K(16x), StrideA, StrideB, StrideC
./bin/example_gemm_mx_fp8 1 1 0 1

# Implies: ./bin/example_gemm_mx_fp8 1 2 0 0
./bin/example_gemm_mx_fp8