From 8e16bebfb11b78647c1e18a888b94d9fbd6c1d89 Mon Sep 17 00:00:00 2001
From: Dongmin Ra <dongmin.ra@moreh.io>
Date: Thu, 2 Oct 2025 15:55:40 +0900
Subject: [PATCH] fixed invalid warpSize in host code

---
 CMakeLists.txt | 10 ++++++++++
 1 file changed, 10 insertions(+)

diff --git a/CMakeLists.txt b/CMakeLists.txt
index bbc9d66a..16bb3ca2 100644
--- a/CMakeLists.txt
+++ b/CMakeLists.txt
@@ -55,6 +55,16 @@ if(USE_ROCM)
   project(mori LANGUAGES HIP CXX C)
   # set(CMAKE_CXX_COMPILER /opt/rocm/bin/hipcc)
   find_package(hip REQUIRED)
+
+  if(DEFINED GPU_TARGETS)
+    list(GET GPU_TARGETS 0 GPU_ARCH)
+    message(STATUS "GPU ARCH: ${GPU_ARCH}")
+    if(GPU_ARCH MATCHES "^gfx8")
+      add_definitions(-D__GFX8__=1)
+    elseif(GPU_ARCH MATCHES "^gfx9")
+      add_definitions(-D__GFX9__=1)
+    endif()
+  endif()
 else()
   set(CUDA_HOME /usr/local/cuda)
   message(STATUS "CUDA_HOME: ${CUDA_HOME}")