[HotFix] Support ROCm 6.1 package build

mlc-ai · Aug 6, 2024 · 87c8a06 · 87c8a06
1 parent ed2e496
commit 87c8a06
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/cmake/modules/ROCM.cmake b/cmake/modules/ROCM.cmake
@@ -23,6 +23,7 @@ if(ROCM_FOUND)
   # avoid global retrigger of cmake
   include_directories(SYSTEM ${ROCM_INCLUDE_DIRS})
   add_definitions(-D__HIP_PLATFORM_HCC__=1)
+  add_definitions(-D__HIP_PLATFORM_AMD__=1)
 endif(ROCM_FOUND)