From 35e652e2865fdcfbce2d7752b352ddf33eb4e838 Mon Sep 17 00:00:00 2001 From: Po Yen Chen Date: Tue, 15 Oct 2024 18:02:41 +0800 Subject: [PATCH] Apply ROCm 6.2 WA to ROCm 6.3 and later (#1563) [ROCm/composable_kernel commit: 9868fd0245eea8f905cd756ab5c6c145c447c596] --- include/ck_tile/core/config.hpp | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/include/ck_tile/core/config.hpp b/include/ck_tile/core/config.hpp index a08c4f3811..a8bc27cdff 100644 --- a/include/ck_tile/core/config.hpp +++ b/include/ck_tile/core/config.hpp @@ -157,8 +157,11 @@ #endif #endif +// workaround for ROCm 6.2 and later #ifndef CK_TILE_WORKAROUND_ROCM_6_2_SCRATCH_MEMORY_ISSUE -#if HIP_VERSION_MAJOR == 6 && HIP_VERSION_MINOR == 2 && HIP_VERSION_PATCH >= 41133 +#if(HIP_VERSION_MAJOR == 6 && HIP_VERSION_MINOR == 2 && HIP_VERSION_PATCH >= 41133) || \ + (HIP_VERSION_MAJOR == 6 && HIP_VERSION_MINOR == 3 && HIP_VERSION_PATCH >= 42131) || \ + (HIP_VERSION_MAJOR == 6 && HIP_VERSION_MINOR > 3) #define CK_TILE_WORKAROUND_ROCM_6_2_SCRATCH_MEMORY_ISSUE 1 #else #define CK_TILE_WORKAROUND_ROCM_6_2_SCRATCH_MEMORY_ISSUE 0