From 318a4933d155807ec7d859060d819af3cefca500 Mon Sep 17 00:00:00 2001 From: Cong Ma <142121551+CongMa13@users.noreply.github.com> Date: Thu, 24 Jul 2025 17:06:32 -0600 Subject: [PATCH] [CK TILE] Apply CK_GFX950_SUPPORT macro on ck tile GEMM unit tests (#2560) cherry-pick c68687e30 and apply CK_GFX950_SUPPORT macro on ck tile GEMM unit tests Co-authored-by: AviralGoelAMD Co-authored-by: Illia Silin <98187287+illsilin@users.noreply.github.com> [ROCm/composable_kernel commit: 2addf05b9116c9d45ce85a3bb1dee15272dd033e] --- CMakeLists.txt | 2 ++ test/ck_tile/gemm/test_gemm_pipeline_smoke_util.hpp | 2 +- 2 files changed, 3 insertions(+), 1 deletion(-) diff --git a/CMakeLists.txt b/CMakeLists.txt index 6e032a30cf..da5a86523e 100644 --- a/CMakeLists.txt +++ b/CMakeLists.txt @@ -236,6 +236,8 @@ endif() if (SUPPORTED_GPU_TARGETS MATCHES "gfx950") add_definitions(-DCK_USE_NATIVE_MX_SUPPORT) set(CK_USE_NATIVE_MX_SUPPORT "ON") + add_definitions(-DCK_GFX950_SUPPORT) + set(CK_GFX950_SUPPORT "ON") endif() option(CK_USE_FP8_ON_UNSUPPORTED_ARCH "Enable FP8 GEMM instances on older architectures" OFF) diff --git a/test/ck_tile/gemm/test_gemm_pipeline_smoke_util.hpp b/test/ck_tile/gemm/test_gemm_pipeline_smoke_util.hpp index bd197150a4..f64d3e092b 100644 --- a/test/ck_tile/gemm/test_gemm_pipeline_smoke_util.hpp +++ b/test/ck_tile/gemm/test_gemm_pipeline_smoke_util.hpp @@ -25,7 +25,7 @@ class ArgumentsNotSupportedException : public std::logic_error template constexpr ck_tile::index_t get_k_warp_tile() { -#if defined(__gfx950__) +#if defined(CK_GFX950_SUPPORT) constexpr bool is_8bit_float = std::is_same_v || std::is_same_v; if constexpr(M_Warp_Tile == 32)