SWDEV-561273 - hip samples on TheRock build using HIP LANGUAGE and hip-lang package. (#1794)

projects/hip-tests/samples/common/DetectGPUArchs.cmake

@@ -0,0 +1,71 @@
+# Copyright (c) 2025 Advanced Micro Devices, Inc. All Rights Reserved.
+#
+# Permission is hereby granted, free of charge, to any person obtaining a copy
+# of this software and associated documentation files (the "Software"), to deal
+# in the Software without restriction, including without limitation the rights
+# to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+# copies of the Software, and to permit persons to whom the Software is
+# furnished to do so, subject to the following conditions:
+#
+# The above copyright notice and this permission notice shall be included in
+# all copies or substantial portions of the Software.
+#
+# THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+# IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+# FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT.  IN NO EVENT SHALL THE
+# AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+# LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+# OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
+# THE SOFTWARE.
+
+# Detect unique GPU architectures for add_custom_command
+if(NOT DEFINED HIP_ARCH_DETECTION_DONE)
+  set(HIP_ARCH_DETECTION_DONE TRUE CACHE INTERNAL "HIP architecture detection completed")
+  
+  # Detect GPU architectures for code object generation
+  # Note: CMake's HIP language support may not deduplicate architectures
+  # Fix any duplicates in CMAKE_HIP_ARCHITECTURES if already set
+  if(DEFINED CMAKE_HIP_ARCHITECTURES AND CMAKE_HIP_ARCHITECTURES)
+    # Convert to list and remove duplicates
+    string(REPLACE ";" ";" GPU_ARCH_LIST "${CMAKE_HIP_ARCHITECTURES}")
+    list(REMOVE_ITEM GPU_ARCH_LIST "gfx000" "")
+    list(REMOVE_DUPLICATES GPU_ARCH_LIST)
+    set(CMAKE_HIP_ARCHITECTURES "${GPU_ARCH_LIST}" CACHE STRING "HIP architectures" FORCE)
+    message(STATUS "Deduplicated HIP architectures: ${CMAKE_HIP_ARCHITECTURES}")
+  elseif(NOT DEFINED CMAKE_HIP_ARCHITECTURES OR NOT CMAKE_HIP_ARCHITECTURES)
+    # Auto-detect if not set
+    if(NOT DEFINED ROCM_PATH)
+      if(DEFINED ENV{ROCM_PATH})
+        set(ROCM_PATH $ENV{ROCM_PATH})
+      else()
+        set(ROCM_PATH "/opt/rocm")
+      endif()
+    endif()
+    
+    execute_process(
+      COMMAND ${ROCM_PATH}/bin/rocm_agent_enumerator
+      OUTPUT_VARIABLE DETECTED_GPUS
+      OUTPUT_STRIP_TRAILING_WHITESPACE
+      ERROR_QUIET
+    )
+    
+    if(DETECTED_GPUS)
+      string(REPLACE "\n" ";" GPU_ARCH_LIST "${DETECTED_GPUS}")
+      list(REMOVE_ITEM GPU_ARCH_LIST "gfx000" "")
+      list(REMOVE_DUPLICATES GPU_ARCH_LIST)
+      set(CMAKE_HIP_ARCHITECTURES "${GPU_ARCH_LIST}" CACHE STRING "HIP architectures" FORCE)
+      message(STATUS "Detected HIP architectures: ${CMAKE_HIP_ARCHITECTURES}")
+    else()
+      set(CMAKE_HIP_ARCHITECTURES "gfx90a" CACHE STRING "HIP architectures" FORCE)
+      message(STATUS "Could not detect GPU, using default: ${CMAKE_HIP_ARCHITECTURES}")
+    endif()
+  endif()
+endif()
+
+# For custom commands that need --offload-arch flags, convert the list to multiple flags
+# This needs to be regenerated each time in case CMAKE_HIP_ARCHITECTURES changed
+set(OFFLOAD_ARCH_FLAGS "")
+foreach(arch ${CMAKE_HIP_ARCHITECTURES})
+  list(APPEND OFFLOAD_ARCH_FLAGS "--offload-arch=${arch}")
+endforeach()
+

projects/hip-tests/samples/common/hip_helper.h

@@ -20,6 +20,8 @@ OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
 THE SOFTWARE.
 */
 
+#include <cstdio>
+#include <cstdlib>
 #include "hip/hip_runtime.h"
 
 #ifndef checkHipErrors