From 6f723d0627064d7e253fc03bf2c17c342dbabbed Mon Sep 17 00:00:00 2001 From: Thomas Ning Date: Fri, 20 Jun 2025 12:50:13 -0700 Subject: [PATCH] fix the mi350 error (#2378) [ROCm/composable_kernel commit: df6023e305f389bbf7249b0c4414e649f3ad6598] --- example/ck_tile/19_gemm_multi_d/CMakeLists.txt | 5 +++++ test/ck_tile/gemm/CMakeLists.txt | 4 ++-- test/ck_tile/gemm_multi_d/CMakeLists.txt | 10 ++++++++-- 3 files changed, 15 insertions(+), 4 deletions(-) diff --git a/example/ck_tile/19_gemm_multi_d/CMakeLists.txt b/example/ck_tile/19_gemm_multi_d/CMakeLists.txt index e2e68b325a..4ecfec7ccf 100644 --- a/example/ck_tile/19_gemm_multi_d/CMakeLists.txt +++ b/example/ck_tile/19_gemm_multi_d/CMakeLists.txt @@ -1 +1,6 @@ add_executable(tile_example_gemm_multi_d_fp16 EXCLUDE_FROM_ALL gemm_multi_d_fp16.cpp) +set(EXAMPLE_GEMM_COMPILE_OPTIONS) +if(CK_USE_OCP_FP8) + list(APPEND EXAMPLE_GEMM_COMPILE_OPTIONS -DCK_TILE_USE_OCP_FP8) +endif() +target_compile_options(tile_example_gemm_multi_d_fp16 PRIVATE ${EXAMPLE_GEMM_COMPILE_OPTIONS}) diff --git a/test/ck_tile/gemm/CMakeLists.txt b/test/ck_tile/gemm/CMakeLists.txt index cfc5b0cd1a..8f880b8fde 100644 --- a/test/ck_tile/gemm/CMakeLists.txt +++ b/test/ck_tile/gemm/CMakeLists.txt @@ -1,9 +1,9 @@ # Currently ck_tile_gemm is only built on gfx94/gfx95 -set(EXAMPLE_GEMM_COMPILE_OPTIONS "") +set(EXAMPLE_GEMM_COMPILE_OPTIONS) if(CK_USE_OCP_FP8) list(APPEND EXAMPLE_GEMM_COMPILE_OPTIONS -DCK_TILE_USE_OCP_FP8) endif() -set(EXAMPLE_GEMM_COMPILE_COMPUTE_V4_OPTIONS "") +set(EXAMPLE_GEMM_COMPILE_COMPUTE_V4_OPTIONS) if(CK_USE_OCP_FP8) list(APPEND EXAMPLE_GEMM_COMPILE_COMPUTE_V4_OPTIONS -DCK_TILE_USE_OCP_FP8) endif() diff --git a/test/ck_tile/gemm_multi_d/CMakeLists.txt b/test/ck_tile/gemm_multi_d/CMakeLists.txt index 1ec77eb87a..a50de7178b 100644 --- a/test/ck_tile/gemm_multi_d/CMakeLists.txt +++ b/test/ck_tile/gemm_multi_d/CMakeLists.txt @@ -1,4 +1,10 @@ # Currently ck_tile is only built on gfx9 -if(GPU_TARGETS MATCHES "gfx9") - add_gtest_executable(test_ck_tile_gemm_multi_d test_gemm_multi_d.cpp) +set(EXAMPLE_GEMM_COMPILE_OPTIONS) +if(CK_USE_OCP_FP8) + list(APPEND EXAMPLE_GEMM_COMPILE_OPTIONS -DCK_TILE_USE_OCP_FP8) +endif() + +if(GPU_TARGETS MATCHES "gfx94" OR GPU_TARGETS MATCHES "gfx95") + add_gtest_executable(test_ck_tile_gemm_multi_d test_gemm_multi_d.cpp) + target_compile_definitions(test_ck_tile_gemm_multi_d PRIVATE ${EXAMPLE_GEMM_COMPILE_OPTIONS}) endif()