Files
clang-p2996/offload/unittests/CMakeLists.txt
Joseph Huber 3cff3d882b [Offload] Add skeleton for offload conformance tests (#146391)
Summary:
This adds a basic outline for adding 'conformance' tests. These are
tests that are intended to check device code against a standard. In this
case, we will expect this to be filled with math conformance tests to
make sure their results are within the ULP requirements we demand.

Right now this just *assumes* the GPU libc is there, meaning you'll
likely need to do a manual `ninja` before doing `ninja -C
runtimes/runtimes-bins offload.conformance`.
2025-07-01 10:20:40 -05:00

115 lines
4.4 KiB
CMake

add_custom_target(OffloadUnitTests)
set_target_properties(OffloadUnitTests PROPERTIES FOLDER "Tests/UnitTests")
function(add_offload_test_device_code test_filename test_name)
set(SRC_PATH ${CMAKE_CURRENT_SOURCE_DIR}/${test_filename})
set(CMAKE_TRY_COMPILE_TARGET_TYPE STATIC_LIBRARY)
# Try to build with support for NVPTX devices.
if("cuda" IN_LIST LIBOMPTARGET_PLUGINS_TO_BUILD)
find_package(CUDAToolkit QUIET)
if(CUDAToolkit_FOUND)
get_filename_component(cuda_path "${CUDAToolkit_BIN_DIR}" DIRECTORY ABSOLUTE)
endif()
check_cxx_compiler_flag(
"--target=nvptx64-nvidia-cuda -march=native --cuda-path=${cuda_path}" PLATFORM_HAS_NVPTX)
if(OFFLOAD_TESTS_FORCE_NVPTX_ARCH)
set(nvptx_arch "${OFFLOAD_TESTS_FORCE_NVPTX_ARCH}")
elseif(PLATFORM_HAS_NVPTX)
set(nvptx_arch "native")
endif()
if(nvptx_arch AND CUDAToolkit_FOUND)
set(output_file "${CMAKE_CURRENT_BINARY_DIR}/${test_name}.nvptx64.bin")
add_custom_command(
OUTPUT ${output_file}
COMMAND ${CMAKE_C_COMPILER}
--target=nvptx64-nvidia-cuda -march=${nvptx_arch}
-nogpulib --cuda-path=${CUDA_ROOT} -flto ${ARGN}
-c ${SRC_PATH} -o ${output_file}
DEPENDS ${SRC_PATH}
)
add_custom_target(${test_name}.nvptx64 DEPENDS ${output_file})
endif()
endif()
# Try to build with support for AMDGPU devices.
if("amdgpu" IN_LIST LIBOMPTARGET_PLUGINS_TO_BUILD)
check_cxx_compiler_flag("--target=amdgcn-amd-amdhsa -mcpu=native" PLATFORM_HAS_AMDGPU)
if(OFFLOAD_TESTS_FORCE_AMDGPU_ARCH)
set(amdgpu_arch "${OFFLOAD_TESTS_FORCE_AMDGPU_ARCH}")
elseif(PLATFORM_HAS_AMDGPU)
set(amdgpu_arch "native")
endif()
if(amdgpu_arch)
set(output_file "${CMAKE_CURRENT_BINARY_DIR}/${test_name}.amdgpu.bin")
add_custom_command(
OUTPUT ${output_file}
COMMAND ${CMAKE_C_COMPILER}
--target=amdgcn-amd-amdhsa -mcpu=${amdgpu_arch}
-nogpulib -flto ${ARGN} -c ${SRC_PATH} -o ${output_file}
DEPENDS ${SRC_PATH}
)
add_custom_target(${test_name}.amdgpu DEPENDS ${output_file})
endif()
endif()
# Create a single dependency target for the device code.
add_custom_target(${test_name}.bin)
if(TARGET ${test_name}.amdgpu)
add_dependencies(${test_name}.bin ${test_name}.amdgpu)
endif()
if(TARGET ${test_name}.nvptx64)
add_dependencies(${test_name}.bin ${test_name}.nvptx64)
endif()
endfunction()
function(add_offload_unittest test_dirname)
set(target_name "${test_dirname}.unittests")
list(TRANSFORM ARGN PREPEND "${CMAKE_CURRENT_SOURCE_DIR}/" OUTPUT_VARIABLE files)
add_unittest(OffloadUnitTests "${target_name}"
${CMAKE_CURRENT_SOURCE_DIR}/common/Environment.cpp
${files})
add_dependencies(${target_name} ${PLUGINS_TEST_COMMON} offload_device_binaries)
target_compile_definitions(${target_name} PRIVATE DEVICE_CODE_PATH="${OFFLOAD_TEST_DEVICE_CODE_PATH}")
target_link_libraries(${target_name} PRIVATE ${PLUGINS_TEST_COMMON})
target_include_directories(${target_name} PRIVATE ${PLUGINS_TEST_INCLUDE})
endfunction()
function(add_conformance_test test_name)
set(target_name "${test_name}.conformance")
list(TRANSFORM ARGN PREPEND "${CMAKE_CURRENT_SOURCE_DIR}/" OUTPUT_VARIABLE files)
if(NOT TARGET libc)
message(WARNING "Cannot run conformance tests without the LLVM C library")
return()
endif()
add_executable(${target_name} ${files})
add_dependencies(${target_name} ${PLUGINS_TEST_COMMON} ${test_name}.bin)
target_compile_definitions(${target_name} PRIVATE DEVICE_CODE_PATH="${CONFORMANCE_TEST_DEVICE_CODE_PATH}")
target_link_libraries(${target_name} PRIVATE ${PLUGINS_TEST_COMMON} libc)
target_include_directories(${target_name} PRIVATE ${PLUGINS_TEST_INCLUDE})
set_target_properties(${target_name} PROPERTIES EXCLUDE_FROM_ALL TRUE)
add_custom_target(offload.conformance.${test_name}
COMMAND $<TARGET_FILE:${target_name}>
DEPENDS ${target_name}
COMMENT "Running conformance test ${test_name}")
add_dependencies(offload.conformance offload.conformance.${test_name})
endfunction()
set(OFFLOAD_TESTS_FORCE_NVPTX_ARCH "" CACHE STRING
"Force building of NVPTX device code for Offload unit tests with the given arch, e.g. sm_61")
set(OFFLOAD_TESTS_FORCE_AMDGPU_ARCH "" CACHE STRING
"Force building of AMDGPU device code for Offload unit tests with the given arch, e.g. gfx1030")
add_subdirectory(OffloadAPI)
add_subdirectory(Conformance)