intel/llvm
1
0
Fork 0
mirror of https://github.com/intel/llvm.git synced 2026-01-19 17:45:07 +08:00
Code Issues Packages Projects Releases Wiki Activity

[libc] Add code for detecting NVIDIA GPUs as well

Browse Source 
Recently the `nvptx-arch` tool was added to address the lack of a
similar tool for detecting locally installed NVIDIA GPUs. This patch
adds similar functionality for the already existing `amdgpu-arch` tool
in libc. These will be used to run tests on the user's system in the
future. On a system with both GPUs installed we will just go with the
first GPU detected. In the future we may want to configure the tests to
run on both of them.

Reviewed By: sivachandra, lntue

Differential Revision: https://reviews.llvm.org/D142776
This commit is contained in:
Joseph Huber
2023-01-27 13:04:50 -06:00
parent 8583291dfc
commit feedb4ad7d
1 changed files with 31 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

33
libc/cmake/modules/prepare_libc_gpu_build.cmake
View File

@@ -29,7 +29,7 @@ if(NOT LLVM_LIBC_FULL_BUILD)
"GPU.")
endif()
# Identify any locally installed GPUs to use for testing.
# Identify any locally installed AMD GPUs on the system to use for testing.
find_program(LIBC_AMDGPU_ARCH
NAMES amdgpu-arch
PATHS ${LLVM_BINARY_DIR}/bin /opt/rocm/llvm/bin/)
@@ -46,4 +46,33 @@ if(LIBC_AMDGPU_ARCH)
set(LIBC_GPU_TARGET_ARCHITECTURE "${arch_string}")
endif()
endif()
# TODO: Check for Nvidia GPUs.
if(LIBC_GPU_TARGET_ARCHITECTURE_IS_AMDGPU)
message(STATUS "Found an installed AMD GPU on the system with target "
"architecture ${LIBC_GPU_TARGET_ARCHITECTURE} ")
return()
endif()
# Identify any locally installed NVIDIA GPUs on the system to use for testing.
find_program(LIBC_NVPTX_ARCH
NAMES nvptx-arch
PATHS ${LLVM_BINARY_DIR}/bin)
if(LIBC_NVPTX_ARCH)
execute_process(COMMAND ${LIBC_NVPTX_ARCH}
OUTPUT_VARIABLE LIBC_NVPTX_ARCH_OUTPUT
OUTPUT_STRIP_TRAILING_WHITESPACE)
string(FIND "${LIBC_NVPTX_ARCH_OUTPUT}" "\n" first_arch_string)
string(SUBSTRING "${LIBC_NVPTX_ARCH_OUTPUT}" 0 ${first_arch_string}
arch_string)
if(arch_string)
set(LIBC_GPU_TARGET_ARCHITECTURE_IS_NVPTX TRUE)
set(LIBC_GPU_TARGET_TRIPLE "nvptx64-nvidia-cuda")
set(LIBC_GPU_TARGET_ARCHITECTURE "${arch_string}")
endif()
endif()
if(LIBC_GPU_TARGET_ARCHITECTURE_IS_NVPTX)
message(STATUS "Found an installed NVIDIA GPU on the system with target "
"architecture ${LIBC_GPU_TARGET_ARCHITECTURE} ")
return()
endif()