| include(External) |
| include(GPUTestVariant) |
| llvm_externals_find(TEST_SUITE_HIP_ROOT "hip" "HIP prerequisites") |
| |
| # Create targets for HIP tests that are part of the test suite. |
| macro(create_local_hip_tests VariantSuffix) |
| set(VariantOffload "hip") |
| # Set per-source compilation/link options |
| set_source_files_properties(with-fopenmp.hip PROPERTIES |
| COMPILE_FLAGS -fopenmp) |
| # Add HIP tests to be added to hip-tests-simple |
| list(APPEND HIP_LOCAL_TESTS empty) |
| list(APPEND HIP_LOCAL_TESTS with-fopenmp) |
| list(APPEND HIP_LOCAL_TESTS saxpy) |
| foreach(_hip_test IN LISTS HIP_LOCAL_TESTS) |
| create_one_local_test(${_hip_test} ${_hip_test}.hip |
| ${VariantOffload} ${VariantSuffix} |
| "${VariantCPPFLAGS}" "${VariantLibs}") |
| endforeach() |
| endmacro() |
| |
| function(create_hip_test VariantSuffix) |
| message(STATUS "Creating HIP test variant ${VariantSuffix}") |
| add_custom_target(hip-tests-simple-${VariantSuffix} |
| COMMENT "Build HIP test variant ${VariantSuffix}") |
| |
| set(VariantCPPFLAGS ${_HIP_CPPFLAGS}) |
| set(VariantLibs ${_HIP_Libs}) |
| list(APPEND LDFLAGS ${_HIP_LDFLAGS}) |
| |
| create_local_hip_tests(${VariantSuffix}) |
| add_dependencies(hip-tests-simple hip-tests-simple-${VariantSuffix}) |
| |
| add_custom_target(check-hip-simple-${VariantSuffix} |
| COMMAND ${TEST_SUITE_LIT} ${TEST_SUITE_LIT_FLAGS} |
| ${VARIANT_SIMPLE_TEST_TARGETS} |
| WORKING_DIRECTORY ${CMAKE_CURRENT_BINARY_DIR} |
| DEPENDS hip-tests-simple-${VariantSuffix} |
| USES_TERMINAL) |
| add_dependencies(check-hip-simple check-hip-simple-${VariantSuffix}) |
| endfunction(create_hip_test) |
| |
| macro(create_hip_tests) |
| # Find all rocm installations at Externals/hip/ directory. |
| # For ROCm, the path looks like rocm-4.1.0 |
| message(STATUS "Checking HIP prerequisites in ${TEST_SUITE_HIP_ROOT}") |
| file(GLOB RocmVersions ${TEST_SUITE_HIP_ROOT}/rocm-*) |
| list(SORT RocmVersions) |
| foreach(RocmDir IN LISTS RocmVersions) |
| get_version(RocmVersion ${RocmDir}) |
| message(STATUS "Found ROCm ${RocmVersion}") |
| list(APPEND ROCM_PATHS ${RocmDir}) |
| add_library(amdhip64-${RocmVersion} SHARED IMPORTED) |
| set_property(TARGET amdhip64-${RocmVersion} PROPERTY IMPORTED_LOCATION |
| ${RocmDir}/lib/libamdhip64.so) |
| endforeach(RocmDir) |
| |
| if(NOT ROCM_PATHS) |
| message(SEND_ERROR |
| "There are no ROCm installations in ${TEST_SUITE_HIP_ROOT}") |
| return() |
| endif() |
| |
| add_custom_target(hip-tests-simple |
| COMMENT "Build all simple HIP tests") |
| add_custom_target(check-hip-simple |
| COMMENT "Run all simple HIP tests") |
| |
| if(NOT AMDGPU_ARCHS) |
| list(APPEND AMDGPU_ARCHS "gfx906;gfx908") |
| endif() |
| |
| foreach(_RocmPath ${ROCM_PATHS}) |
| get_version(_RocmVersion ${_RocmPath}) |
| set(_HIP_Suffix "hip-${_RocmVersion}") |
| # Set up HIP test flags |
| set(_HIP_CPPFLAGS -D__HIP_ROCclr__ -xhip --hip-device-lib-path=${_RocmPath}/amdgcn/bitcode |
| -I${_RocmPath}/include) |
| set(_HIP_LDFLAGS -L${_RocmPath}/lib -lamdhip64) |
| set(_HIP_Libs amdhip64-${RocmVersion}) |
| |
| # Unset these for each iteration of rocm path. |
| set(_ArchFlags) |
| set(_ArchList) |
| foreach(_AMDGPUArch IN LISTS AMDGPU_ARCHS) |
| list(APPEND _ArchFlags --offload-arch=${_AMDGPUArch}) |
| endforeach() |
| message(STATUS "Building ${_RocmPath} targets for ${AMDGPU_ARCHS}") |
| list(APPEND _HIP_CPPFLAGS ${_ArchFlags}) |
| |
| create_hip_test(${_HIP_Suffix}) |
| endforeach() |
| |
| add_custom_target(hip-tests-all DEPENDS hip-tests-simple |
| COMMENT "Build all HIP tests.") |
| |
| file(COPY lit.local.cfg DESTINATION "${CMAKE_CURRENT_BINARY_DIR}") |
| endmacro(create_hip_tests) |
| |
| if(TEST_SUITE_HIP_ROOT) |
| create_hip_tests() |
| endif() |