Skip to content

Commit 6b69ac9

Browse files
[rocm-libraries] ROCm/rocm-libraries#5625 (commit 7d2ed43)
[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.
1 parent a66047a commit 6b69ac9

2 files changed

Lines changed: 21 additions & 7 deletions

File tree

test/ck_tile/gemm_streamk_tile_engine/CMakeLists.txt

Lines changed: 6 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -280,14 +280,18 @@ message(STATUS "Building StreamK GEMM tile engine tests for GPU targets: ${GEMM_
280280
# All supported data types and layouts for comprehensive testing
281281
# Note: fp64 not included (no MFMA hardware support)
282282
set(TEST_DATATYPES "fp16;bf16")
283-
set(TEST_LAYOUTS "rcr;rrr;ccr;crr")
283+
# Temporarily only test rcr and crr
284+
# set(TEST_LAYOUTS "rcr;rrr;ccr;crr")
285+
set(TEST_LAYOUTS "rcr;crr")
284286

285287
# ============================================================================
286288
# Test Target Generation - Datatype-Specific Categories
287289
# ============================================================================
288290

289291
# 1. SMOKE TESTS: Test for basic functionality with data types (fp8, bf8, fp16, bf16)
290-
set(SMALL_DATATYPES "fp16;bf16;fp8;bf8")
292+
# Temporarily only consider fp16
293+
# set(SMALL_DATATYPES "fp16;bf16;fp8;bf8")
294+
set(SMALL_DATATYPES "fp16")
291295
set(SIXTEEN_BIT_DATATYPES "fp16;bf16")
292296
set(EIGHT_BIT_DATATYPES "fp8;bf8")
293297
set(LARGE_TILES "256,256,32")

test/ck_tile/gemm_streamk_tile_engine/generate_configs.py

Lines changed: 15 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -23,7 +23,9 @@ class TileConfig:
2323
warp_k: List[int] = field(default_factory=lambda: [1])
2424
warp_tile_m: List[int] = field(default_factory=lambda: [16, 32])
2525
warp_tile_n: List[int] = field(default_factory=lambda: [16, 32])
26-
warp_tile_k: List[int] = field(default_factory=lambda: [8, 16, 32])
26+
# Temporarily only consider 16 for warp_tile_k
27+
# warp_tile_k: List[int] = field(default_factory=lambda: [8, 16, 32])
28+
warp_tile_k: List[int] = field(default_factory=lambda: [16])
2729

2830
def to_dict(self) -> Dict:
2931
return {k: {"values": v} for k, v in asdict(self).items()}
@@ -33,7 +35,9 @@ def to_dict(self) -> Dict:
3335
class TraitConfig:
3436
"""Represents the Trait Config section of a Tile Engine config"""
3537

36-
pipeline: List[str] = field(default_factory=lambda: ["compv3", "mem"])
38+
# Temporarily only consider compv3
39+
# pipeline: List[str] = field(default_factory=lambda: ["compv3", "mem"])
40+
pipeline: List[str] = field(default_factory=lambda: ["compv3"])
3741
epilogue: List[str] = field(default_factory=lambda: ["cshuffle"])
3842
scheduler: List[str] = field(default_factory=lambda: ["intrawave"])
3943
pad_m: List[bool] = field(default_factory=lambda: [False])
@@ -67,21 +71,27 @@ def __init__(
6771
0,
6872
["atomic"],
6973
[True, False],
70-
["fp16", "bf16", "fp8", "bf8"],
74+
# Temporarily only run fp16 tests
75+
# ["fp16", "bf16", "fp8", "bf8"],
76+
["fp16"],
7177
"Stream-K atomic smoke tests",
7278
)
7379
REDUCTION_SMOKE = (
7480
2,
7581
["linear", "tree"],
7682
[True, False],
77-
["fp16", "bf16", "fp8", "bf8"],
83+
# Temporarily only run fp16 tests
84+
# ["fp16", "bf16", "fp8", "bf8"],
85+
["fp16"],
7886
"Stream-K reduction smoke tests",
7987
)
8088
EXTENDED = (
8189
3,
8290
["atomic"],
8391
[True, False],
84-
["fp16", "bf16", "fp8", "bf8"],
92+
# Temporarily only run fp16 tests
93+
# ["fp16", "bf16", "fp8", "bf8"],
94+
["fp16"],
8595
"Stream-K extended smoke tests",
8696
)
8797

0 commit comments

Comments
 (0)