Enable scratch memory workaround on ROCm 6.2 (#1486)

Co-authored-by: carlushuang <carlus.huang@amd.com>
This commit is contained in:
Po Yen Chen
2024-08-30 10:40:00 +08:00
committed by GitHub
parent 8107ee6270
commit 461ec98d78
2 changed files with 10 additions and 1 deletions

View File

@@ -156,6 +156,14 @@
#endif
#endif
#ifndef CK_TILE_WORKAROUND_ROCM_6_2_SCRATCH_MEMORY_ISSUE
#if HIP_VERSION_MAJOR == 6 && HIP_VERSION_MINOR == 2 && HIP_VERSION_PATCH >= 41133
#define CK_TILE_WORKAROUND_ROCM_6_2_SCRATCH_MEMORY_ISSUE 1
#else
#define CK_TILE_WORKAROUND_ROCM_6_2_SCRATCH_MEMORY_ISSUE 0
#endif
#endif
#ifndef CK_TILE_DEBUG_LOG
#define CK_TILE_DEBUG_LOG 0
#endif