| # Copyright 2023 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 |
| |
| set(IREE_PACKAGE_ROOT_DIR "${CMAKE_CURRENT_LIST_DIR}") |
| set(IREE_PACKAGE_ROOT_PREFIX "") |
| |
| add_library(iree_pjrt_compiler_defs INTERFACE) |
| target_include_directories(iree_pjrt_compiler_defs |
| INTERFACE |
| ${CMAKE_CURRENT_SOURCE_DIR} |
| ${CMAKE_CURRENT_BINARY_DIR} |
| ) |
| |
| # Configures all iree_cc_* targets to take this implicit dep, |
| # which provides common includes and copts for the tree. |
| set(IREE_IMPLICIT_DEFS_CC_DEPS iree_pjrt_compiler_defs) |
| |
| add_subdirectory(iree_pjrt/common) |
| add_subdirectory(iree_pjrt/partitioner_api) |
| |
| if(IREE_HAL_DRIVER_LOCAL_TASK) |
| add_subdirectory(iree_pjrt/cpu) |
| endif() |
| if(IREE_HAL_DRIVER_CUDA) |
| add_subdirectory(iree_pjrt/cuda) |
| endif() |
| if("rocm" IN_LIST IREE_EXTERNAL_HAL_DRIVERS) |
| add_subdirectory(iree_pjrt/rocm) |
| endif() |
| if(IREE_HAL_DRIVER_VULKAN) |
| add_subdirectory(iree_pjrt/vulkan) |
| endif() |
