From 30ab79d84f518c6dbea14ec6047cd6442b76e3e8 Mon Sep 17 00:00:00 2001 From: Aviral Goel Date: Mon, 25 Aug 2025 21:53:40 -0400 Subject: [PATCH] fix(gemm_universal): define CK_TILE_USE_WMMA with default value to stop compilation error (#2737) [ROCm/composable_kernel commit: c88e24ebe5e929b62373cc83106bd89879a6a915] --- CMakeLists.txt | 11 +++++++++-- 1 file changed, 9 insertions(+), 2 deletions(-) diff --git a/CMakeLists.txt b/CMakeLists.txt index f148f31d25..39eb815680 100644 --- a/CMakeLists.txt +++ b/CMakeLists.txt @@ -221,13 +221,20 @@ if (SUPPORTED_GPU_TARGETS MATCHES "gfx94" OR SUPPORTED_GPU_TARGETS MATCHES "gfx9 add_definitions(-DCK_USE_GFX94) set(CK_USE_GFX94 "ON") endif() + +# new macro CK_TILE_USE_WMMA in order to separately compile examples for MFMA/WMMA +set(CK_TILE_USE_WMMA 0) + if (SUPPORTED_GPU_TARGETS MATCHES "gfx11" OR SUPPORTED_GPU_TARGETS MATCHES "gfx12") message(STATUS "Enabling WMMA instances") add_definitions(-DCK_USE_WMMA) set(CK_USE_WMMA "ON") - add_definitions(-DCK_TILE_USE_WMMA) - set(CK_TILE_USE_WMMA "ON") + set(CK_TILE_USE_WMMA 1) endif() + +# define the macro with the current value (0 or 1) +add_definitions(-DCK_TILE_USE_WMMA=${CK_TILE_USE_WMMA}) + if (SUPPORTED_GPU_TARGETS MATCHES "gfx12") message(STATUS "Enabling WMMA FP8 gemms on native architectures") add_definitions(-DCK_USE_WMMA_FP8)