Files
composable_kernel/test
Yung-sheng Tu 8c5870f962 [rocm-libraries] ROCm/rocm-libraries#8020 (commit fa9cdd1)
=?UTF-8?q?feat:=20[CK=20TILE]=20Unification=20Work=20?=
 =?UTF-8?q?=E2=80=93=20Add=20WMMA=20Scale=20Mixed=20Types=20Support=20(#80?=
 =?UTF-8?q?20)?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

## Motivation

This PR adds mixed data types and mixed scale types support in scale
WMMA specialisations.

## Technical Details

- Replaced `ScaleDataTypeToFlag_v` with `PackedDataTypeToFlag_v` and
adjusted WMMA scale selector to the way as MFMA one.

- Simplified scale_gfx125.hpp with macros.

- Added WMMA scale specialisations with mixed data types and scale
types.

- Introduced `is_legal_combination` to restrict which combinations of
ADataType/BDataType/ScaleAValueType/ScaleBValueType the `exec()` can be
instantiated.

- Pleased the compiler on gfx1250 emulator to make compilation working.

## Test Plan

All the new wrappers were added to the test suite in
test_amdgcn_mma_layout.inc.

## Test Result

Test should pass.

## Submission Checklist

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

close #8949
2026-07-14 20:51:36 +00:00
..
2025-12-16 19:50:49 -08:00