# 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()