Added ability to build for local gpu by env Variable (#204)

* Added the ability to compile for Local gpu by environment variable

* adding gfx950 on default only on rocm 7.0 and above

* Updated docs

* removed xnack+ on specific gfx targets

---------

Co-authored-by: Yiltan Hassan Temucin <yiltan.temucin@amd.com>
This commit is contained in:
akolliasAMDandYiltan Hassan Temucin authored and GitHub committed 2025-08-11 12:35:50 -06:00
1 parent e95360961d
commit be630d9b93
4 files changed
+45 -5

No files matched your search

+31
View File
@@ -31,6 +31,37 @@ set(CMAKE_BUILD_TYPE "Release" CACHE STRING
###############################################################################
# GLOBAL COMPILE FLAGS
###############################################################################
# Try to establish ROCM_PATH (for find_package)
#==================================================================================================
if(NOT DEFINED ROCM_PATH)
# Guess default location
set(ROCM_PATH "/opt/rocm")
message(WARNING "Unable to find ROCM_PATH: Falling back to ${ROCM_PATH}")
else()
message(STATUS "ROCM_PATH found: ${ROCM_PATH}")
endif()
set(ENV{ROCM_PATH} ${ROCM_PATH})
## Check for ROCm version
if(ROCM_PATH)
message(STATUS "Reading ROCM version from ${ROCM_PATH}/.info/version")
file(READ "${ROCM_PATH}/.info/version" rocm_version_string)
else()
message(FATAL_ERROR "Could not determine ROCM version (set EXPLICIT_ROCM_VERSION or set ROCM_PATH to a valid installation)")
endif()
string(REGEX MATCH "([0-9]+)\\.([0-9]+)\\.([0-9]+)" rocm_version_matches ${rocm_version_string})
if (rocm_version_matches)
set(ROCM_MAJOR_VERSION ${CMAKE_MATCH_1})
set(ROCM_MINOR_VERSION ${CMAKE_MATCH_2})
set(ROCM_PATCH_VERSION ${CMAKE_MATCH_3})
message(STATUS "ROCm version: ${ROCM_MAJOR_VERSION}.${ROCM_MINOR_VERSION}.${ROCM_PATCH_VERSION}")
else()
message(WARNING "Failed to extract ROCm version.")
endif()
foreach (root ${hip_ROOT} $ENV{hip_ROOT} ${ROCM_ROOT} $ENV{ROCM_ROOT} ${ROCM_PATH} $ENV{ROCM_PATH})
if (IS_DIRECTORY ${root})
list(PREPEND CMAKE_PREFIX_PATH ${root})