set(IREE_RUNTIME_SOURCE_DIR "${IREE_SOURCE_DIR}/runtime/src/iree")

iree_cc_library(
  NAME
    arch
  HDRS
    "${IREE_RUNTIME_SOURCE_DIR}/builtins/ukernel/arch/mmt4d_arch.h"
    "${IREE_RUNTIME_SOURCE_DIR}/builtins/ukernel/arch/pack_arch.h"
    "${IREE_RUNTIME_SOURCE_DIR}/builtins/ukernel/arch/query_tile_sizes_arch.h"
  SRCS
    "mmt4d_arch.c"
    "pack_arch.c"
    "query_tile_sizes_arch.c"
  DEPS
    iree::builtins::ukernel::common
    iree::builtins::ukernel::generic
  PUBLIC
)

iree_cc_library(
  NAME
    ukernel
  HDRS
    "${IREE_RUNTIME_SOURCE_DIR}/builtins/ukernel/elementwise.h"
    "${IREE_RUNTIME_SOURCE_DIR}/builtins/ukernel/mmt4d.h"
    "${IREE_RUNTIME_SOURCE_DIR}/builtins/ukernel/pack.h"
    "${IREE_RUNTIME_SOURCE_DIR}/builtins/ukernel/query_tile_sizes.h"
    "${IREE_RUNTIME_SOURCE_DIR}/builtins/ukernel/unpack.h"
  SRCS
    "${IREE_RUNTIME_SOURCE_DIR}/builtins/ukernel/mmt4d.c"
    "${IREE_RUNTIME_SOURCE_DIR}/builtins/ukernel/pack.c"
    "${IREE_RUNTIME_SOURCE_DIR}/builtins/ukernel/query_tile_sizes.c"
    "${IREE_RUNTIME_SOURCE_DIR}/builtins/ukernel/unpack.c"
  DEPS
    ::arch
    iree::builtins::ukernel::common
    iree::builtins::ukernel::elementwise
  PUBLIC
)

iree_cc_library(
  NAME
    vmvx
  HDRS
    "${IREE_RUNTIME_SOURCE_DIR}/modules/vmvx/module.h"
  TEXTUAL_HDRS
    "${IREE_RUNTIME_SOURCE_DIR}/modules/vmvx/exports.inl"
  SRCS
    "${IREE_RUNTIME_SOURCE_DIR}/modules/vmvx/module.c"
  DEFINES
    "IREE_HAVE_VMVX_MODULE"
  DEPS
    ::ukernel
  PUBLIC
)

iree_cc_library(
  NAME
    vmvx_module_loader
  HDRS
    "${IREE_RUNTIME_SOURCE_DIR}/hal/local/loaders/vmvx_module_loader.h"
  SRCS
    "${IREE_RUNTIME_SOURCE_DIR}/hal/local/loaders/vmvx_module_loader.c"
  DEPS
    ::vmvx
    iree::hal::local::executable_loader
    iree::vm::bytecode_module
  DEFINES
    "IREE_HAVE_HAL_EXECUTABLE_LOADER_VMVX_MODULE=1"
  PUBLIC
)