ROCm/composable_kernel
1
0
Fork 0
mirror of https://github.com/ROCm/composable_kernel.git synced 2026-05-03 13:11:25 +00:00
Code Issues Packages Projects Releases Wiki Activity

[rocm-libraries] ROCm/rocm-libraries#5625 (commit 7d2ed43)

Browse Source 
[CK_TILE] Prune Stream-K Tile Engine Tests

## Motivation

Stream-K tile engine tests are causing issues for build time. While we
work on a more permanent solution, these changes prune the Stream-K test
instances to help reduce the build time burden.

## Technical Details

The Stream-K team recently transitioned to using CK Tile's tile engine
infrastructure for our smoke tests. However, since tile engine creates
an individual target per kernel instance, we've found that the tile
engine tests are increasing build times. Our team is currently working
to convert our existing tile engine tests back to basic gtests. While
this work takes place, we are temporarily pruning the existing Stream-K
tile engine test instances to help reduce the build time burden.

## Test Plan

Ran the pruned test set on all gfx90a, gfx942, and gfx950.

## Test Result

All tests pass.

## Submission Checklist

- [x] Look over the contributing guidelines at
https://github.com/ROCm/ROCm/blob/develop/CONTRIBUTING.md#pull-requests.
This commit is contained in:
Emily Martins
2026-03-20 20:31:39 +00:00
committed by assistant-librarian[bot]
parent a66047ad09
commit 6b69ac9676
2 changed files with 21 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
test/ck_tile/gemm_streamk_tile_engine/CMakeLists.txt
View File

@@ -280,14 +280,18 @@ message(STATUS "Building StreamK GEMM tile engine tests for GPU targets: ${GEMM_
# All supported data types and layouts for comprehensive testing
# Note: fp64 not included (no MFMA hardware support)
set(TEST_DATATYPES "fp16;bf16")
set(TEST_LAYOUTS "rcr;rrr;ccr;crr")
# Temporarily only test rcr and crr
# set(TEST_LAYOUTS "rcr;rrr;ccr;crr")
set(TEST_LAYOUTS "rcr;crr")
# ============================================================================
# Test Target Generation - Datatype-Specific Categories
# ============================================================================
# 1. SMOKE TESTS: Test for basic functionality with data types (fp8, bf8, fp16, bf16)
set(SMALL_DATATYPES "fp16;bf16;fp8;bf8")
# Temporarily only consider fp16
# set(SMALL_DATATYPES "fp16;bf16;fp8;bf8")
set(SMALL_DATATYPES "fp16")
set(SIXTEEN_BIT_DATATYPES "fp16;bf16")
set(EIGHT_BIT_DATATYPES "fp8;bf8")
set(LARGE_TILES "256,256,32")

20
test/ck_tile/gemm_streamk_tile_engine/generate_configs.py
View File

@@ -23,7 +23,9 @@ class TileConfig:
warp_k: List[int] = field(default_factory=lambda: [1])
warp_tile_m: List[int] = field(default_factory=lambda: [16, 32])
warp_tile_n: List[int] = field(default_factory=lambda: [16, 32])
warp_tile_k: List[int] = field(default_factory=lambda: [8, 16, 32])
# Temporarily only consider 16 for warp_tile_k
# warp_tile_k: List[int] = field(default_factory=lambda: [8, 16, 32])
warp_tile_k: List[int] = field(default_factory=lambda: [16])
def to_dict(self) -> Dict:
return {k: {"values": v} for k, v in asdict(self).items()}
@@ -33,7 +35,9 @@ class TileConfig:
class TraitConfig:
"""Represents the Trait Config section of a Tile Engine config"""
pipeline: List[str] = field(default_factory=lambda: ["compv3", "mem"])
# Temporarily only consider compv3
# pipeline: List[str] = field(default_factory=lambda: ["compv3", "mem"])
pipeline: List[str] = field(default_factory=lambda: ["compv3"])
epilogue: List[str] = field(default_factory=lambda: ["cshuffle"])
scheduler: List[str] = field(default_factory=lambda: ["intrawave"])
pad_m: List[bool] = field(default_factory=lambda: [False])
@@ -67,21 +71,27 @@ class TestVariant(Enum):
0,
["atomic"],
[True, False],
["fp16", "bf16", "fp8", "bf8"],
# Temporarily only run fp16 tests
# ["fp16", "bf16", "fp8", "bf8"],
["fp16"],
"Stream-K atomic smoke tests",
)
REDUCTION_SMOKE = (
2,
["linear", "tree"],
[True, False],
["fp16", "bf16", "fp8", "bf8"],
# Temporarily only run fp16 tests
# ["fp16", "bf16", "fp8", "bf8"],
["fp16"],
"Stream-K reduction smoke tests",
)
EXTENDED = (
3,
["atomic"],
[True, False],
["fp16", "bf16", "fp8", "bf8"],
# Temporarily only run fp16 tests
# ["fp16", "bf16", "fp8", "bf8"],
["fp16"],
"Stream-K extended smoke tests",
)