# 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 | |
if(NOT ${IREE_HAL_DRIVER_EXPERIMENTAL_ROCM}) | |
return() | |
endif() | |
iree_add_all_subdirs() | |
iree_cc_library( | |
NAME | |
rocm | |
HDRS | |
"api.h" | |
SRCS | |
"api.h" | |
"context_wrapper.h" | |
"rocm_allocator.c" | |
"rocm_allocator.h" | |
"rocm_buffer.c" | |
"rocm_buffer.h" | |
"rocm_device.c" | |
"rocm_device.h" | |
"rocm_driver.c" | |
"rocm_event.c" | |
"rocm_event.h" | |
"descriptor_set_layout.c" | |
"descriptor_set_layout.h" | |
"event_semaphore.c" | |
"event_semaphore.h" | |
"executable_layout.c" | |
"executable_layout.h" | |
"direct_command_buffer.c" | |
"direct_command_buffer.h" | |
"native_executable.c" | |
"native_executable.h" | |
"nop_executable_cache.c" | |
"nop_executable_cache.h" | |
"status_util.c" | |
"status_util.h" | |
INCLUDES | |
"${CMAKE_CURRENT_LIST_DIR}/../.." | |
"${PROJECT_BINARY_DIR}" | |
DEPS | |
::dynamic_symbols | |
iree::base | |
iree::base::core_headers | |
iree::base::internal | |
iree::base::internal::arena | |
iree::base::internal::flatcc::parsing | |
iree::base::internal::synchronization | |
iree::base::tracing | |
iree::hal | |
iree::hal::utils::buffer_transfer | |
iree::schemas::rocm_executable_def_c_fbs | |
PUBLIC | |
) | |
add_definitions(-D__HIP_PLATFORM_HCC__) | |
iree_cc_library( | |
NAME | |
dynamic_symbols | |
HDRS | |
"dynamic_symbols.h" | |
TEXTUAL_HDRS | |
"dynamic_symbol_tables.h" | |
SRCS | |
"rocm_headers.h" | |
"dynamic_symbols.c" | |
INCLUDES | |
"${CMAKE_CURRENT_LIST_DIR}/../.." | |
DEPS | |
rocm_headers | |
iree::base::core_headers | |
iree::base::internal::dynamic_library | |
iree::base::tracing | |
PUBLIC | |
) | |
iree_cc_test( | |
NAME | |
dynamic_symbols_test | |
SRCS | |
"dynamic_symbols_test.cc" | |
DEPS | |
::dynamic_symbols | |
iree::base | |
iree::testing::gtest | |
iree::testing::gtest_main | |
LABELS | |
"driver=rocm" | |
) | |
### BAZEL_TO_CMAKE_PRESERVES_ALL_CONTENT_BELOW_THIS_LINE ### |