From 59e6fc67d84ff36d99eae053b216da7b457ca70b Mon Sep 17 00:00:00 2001
From: Emily Martins <65371150+ecamartins@users.noreply.github.com>
Date: Fri, 20 Mar 2026 14:30:34 -0600
Subject: [PATCH] [CK_TILE] Prune Stream-K Tile Engine Tests (#5625)

## 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.
---
 .../gemm_streamk_tile_engine/CMakeLists.txt   |  8 ++++++--
 .../generate_configs.py                       | 20 ++++++++++++++-----
 2 files changed, 21 insertions(+), 7 deletions(-)

diff --git a/test/ck_tile/gemm_streamk_tile_engine/CMakeLists.txt b/test/ck_tile/gemm_streamk_tile_engine/CMakeLists.txt
index aa1a2d2d1c..4acab26c41 100644
--- a/test/ck_tile/gemm_streamk_tile_engine/CMakeLists.txt
+++ b/test/ck_tile/gemm_streamk_tile_engine/CMakeLists.txt
@@ -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")
diff --git a/test/ck_tile/gemm_streamk_tile_engine/generate_configs.py b/test/ck_tile/gemm_streamk_tile_engine/generate_configs.py
index 0f2673c6dd..2795303684 100644
--- a/test/ck_tile/gemm_streamk_tile_engine/generate_configs.py
+++ b/test/ck_tile/gemm_streamk_tile_engine/generate_configs.py
@@ -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",
     )