composable_kernel

mirror of https://github.com/ROCm/composable_kernel.git synced 2026-05-11 17:00:18 +00:00

Files

Max Podkorytov 0438ab1b79 [rocm-libraries] ROCm/rocm-libraries#4518 (commit dd161dc)

[CK_TILE] Fix CShuffleEpilogue test to use correct GEMM
 accumulator distribution (#4518)
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

## Summary

The test was using LDS distribution to create the accumulator tile, but
CShuffleEpilogue expects the GEMM accumulator distribution that
BlockGemm produces. This mismatch caused incorrect data permutation.

## Changes

- Use WarpGemmDispatcher to get correct accumulator distribution
encoding
- Load test input from host-initialized global memory for deterministic
verification
- Shard tests by data type (FP16, FP8) with gfx950-specific FP8 tests
- Extract scale tests into separate target for better organization
- Implement exact permutation verification (all unique values appear
once)
- Reduce tile size from 256x256 to 128x128 to fit in unique fp16 range
- Add parameterized test configurations for various warp layouts and
MFMA types

## Test plan

- [x] Run new cshuffle epilogue tests

🤖 Generated with [Claude Code](https://claude.com/claude-code)

Co-Authored-By: Claude <noreply@anthropic.com>

2026-03-02 08:55:05 +00:00

CMakeLists.txt

[rocm-libraries] ROCm/rocm-libraries#4518 (commit dd161dc)

2026-03-02 08:55:05 +00:00

test_cshuffle_epilogue_common.hpp

[rocm-libraries] ROCm/rocm-libraries#4518 (commit dd161dc)

2026-03-02 08:55:05 +00:00

test_cshuffle_epilogue_fp8_gfx950.cpp

[rocm-libraries] ROCm/rocm-libraries#4518 (commit dd161dc)

2026-03-02 08:55:05 +00:00

test_cshuffle_epilogue_fp8.cpp

[rocm-libraries] ROCm/rocm-libraries#4518 (commit dd161dc)