| # Copyright 2021 The IREE Authors |
| # |
| # Licensed under the Apache License v2.0 with LLVM Exceptions. |
| # See https://llvm.org/LICENSE.txt for license information. |
| # SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception |
| |
| unset(COMPILER_FLAGS) |
| if(IREE_ROCM_LINK_BC) |
| list(APPEND COMPILER_FLAGS "--iree-rocm-link-bc=true") |
| else() |
| list(APPEND COMPILER_FLAGS "--iree-rocm-link-bc=false") |
| endif() |
| list(APPEND COMPILER_FLAGS |
| "--iree-rocm-target-chip=${IREE_ROCM_TARGET_CHIP}" |
| "--iree-rocm-bc-dir=${IREE_ROCM_BC_DIR}") |
| |
| iree_hal_cts_test_suite( |
| DRIVER_NAME |
| rocm |
| DRIVER_REGISTRATION_HDR |
| "experimental/rocm/registration/driver_module.h" |
| DRIVER_REGISTRATION_FN |
| "iree_hal_rocm_driver_module_register" |
| COMPILER_TARGET_BACKEND |
| "rocm" |
| EXECUTABLE_FORMAT |
| "\"PTXE\"" |
| COMPILER_FLAGS |
| ${COMPILER_FLAGS} |
| DEPS |
| iree::experimental::rocm::registration |
| EXCLUDED_TESTS |
| # This test depends on iree_hal_rocm_direct_command_buffer_update_buffer |
| # via iree_hal_buffer_view_allocate_buffer_copy, which is not implemented yet. |
| "command_buffer_dispatch" |
| # Semaphores are not implemented in the ROCm backend yet. |
| "semaphore_submission" |
| "semaphore" |
| LABELS |
| driver=rocm |
| ) |
