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:
1 parent
e95360961d
commit
be630d9b93
4 files changed
+45
-5
No files matched your search
@@ -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})
|
||||
|
||||
Reference in new issue
Block a user