iree/vm/CMakeLists.txt

# Copyright 2019 Google LLC
#
# Licensed under the Apache License, Version 2.0 (the "License");
# you may not use this file except in compliance with the License.
# You may obtain a copy of the License at
#
#      https://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS,
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and
# limitations under the License.

iree_add_all_subdirs()

iree_cc_library(
  NAME
    builtin_types
  HDRS
    "builtin_types.h"
  SRCS
    "builtin_types.c"
  DEPS
    ::list
    ::ref
    iree::base::api
  PUBLIC
)

iree_cc_test(
  NAME
    bytecode_dispatch_test
  SRCS
    "bytecode_dispatch_test.cc"
  DEPS
    ::builtin_types
    ::bytecode_module
    ::context
    ::instance
    ::invocation
    ::module
    absl::strings
    iree::base::logging
    iree::base::status
    iree::testing::gtest
    iree::testing::gtest_main
    iree::vm::test::all_bytecode_modules_cc
)

iree_cc_library(
  NAME
    bytecode_module
  HDRS
    "bytecode_module.h"
  SRCS
    "bytecode_dispatch.c"
    "bytecode_dispatch_util.h"
    "bytecode_module.c"
    "bytecode_module_impl.h"
    "bytecode_op_table.h"
  DEPS
    ::builtin_types
    ::list
    ::module
    ::ref
    ::stack
    ::type_def
    ::value
    flatcc::runtime
    iree::base::alignment
    iree::base::api
    iree::base::tracing
    iree::schemas::bytecode_module_def_c_fbs
  PUBLIC
)

iree_cc_test(
  NAME
    bytecode_module_benchmark
  SRCS
    "bytecode_module_benchmark.cc"
  DEPS
    ::bytecode_module
    ::bytecode_module_benchmark_module_cc
    ::context
    ::instance
    ::module
    ::native_module
    ::stack
    absl::inlined_vector
    absl::strings
    benchmark
    iree::base::api
    iree::base::logging
    iree::testing::benchmark_main
)

iree_bytecode_module(
  NAME
    bytecode_module_benchmark_module
  SRC
    "bytecode_module_benchmark.mlir"
  CC_NAMESPACE
    "iree::vm"
  FLAGS
    "-iree-vm-ir-to-bytecode-module"
  PUBLIC
)

iree_cc_test(
  NAME
    bytecode_module_size_benchmark
  SRCS
    "bytecode_module_size_benchmark.cc"
  DEPS
    ::bytecode_module
    ::bytecode_module_size_benchmark_module_cc
    ::vm
    iree::base::api
)

iree_bytecode_module(
  NAME
    bytecode_module_size_benchmark_module
  SRC
    "bytecode_module_size_benchmark.mlir"
  CC_NAMESPACE
    "iree::vm"
  FLAGS
    "-iree-vm-ir-to-bytecode-module"
  PUBLIC
)

iree_cc_test(
  NAME
    bytecode_module_test
  SRCS
    "bytecode_module_test.cc"
  DEPS
    ::bytecode_module
    iree::testing::gtest
    iree::testing::gtest_main
)

iree_tablegen_library(
  NAME
    bytecode_op_table_gen
  TD_FILE
    "${IREE_ROOT_DIR}/iree/compiler/Dialect/VM/IR/VMOps.td"
  OUTS
    -gen-iree-vm-op-table-defs bytecode_op_table.h
  TBLGEN
    IREE
)

iree_cc_library(
  NAME
    c_funcs
  HDRS
    "c_funcs.h"
  PUBLIC
)

iree_cc_library(
  NAME
    context
  HDRS
    "context.h"
  SRCS
    "context.c"
  DEPS
    ::instance
    ::module
    ::stack
    iree::base::api
    iree::base::atomics
    iree::base::tracing
  PUBLIC
)

iree_cc_library(
  NAME
    instance
  HDRS
    "instance.h"
  SRCS
    "instance.c"
  DEPS
    ::builtin_types
    iree::base::api
    iree::base::atomics
    iree::base::tracing
  PUBLIC
)

iree_cc_library(
  NAME
    invocation
  HDRS
    "invocation.h"
  SRCS
    "invocation.c"
  DEPS
    ::context
    ::list
    ::module
    iree::base::api
    iree::base::tracing
  PUBLIC
)

iree_cc_library(
  NAME
    list
  HDRS
    "list.h"
  SRCS
    "list.c"
  DEPS
    ::ref
    ::type_def
    ::value
    iree::base::alignment
    iree::base::api
  PUBLIC
)

iree_cc_test(
  NAME
    list_test
  SRCS
    "list_test.cc"
  DEPS
    ::builtin_types
    ::list
    iree::base::api
    iree::base::ref_ptr
    iree::testing::gtest
    iree::testing::gtest_main
)

iree_cc_library(
  NAME
    module
  HDRS
    "module.h"
  SRCS
    "module.c"
  DEPS
    ::ref
    iree::base::alignment
    iree::base::api
    iree::base::atomics
    iree::base::tracing
  PUBLIC
)

iree_cc_library(
  NAME
    native_module
  HDRS
    "native_module.h"
  SRCS
    "native_module.c"
  DEPS
    ::module
    ::stack
    iree::base::api
  PUBLIC
)

iree_cc_test(
  NAME
    native_module_benchmark
  SRCS
    "native_module_benchmark.cc"
  DEPS
    ::module
    ::native_module
    ::native_module_test_hdrs
    ::stack
    benchmark
    iree::base::api
    iree::base::logging
    iree::testing::benchmark_main
)

iree_cc_test(
  NAME
    native_module_test
  SRCS
    "native_module_test.cc"
  DEPS
    ::context
    ::instance
    ::invocation
    ::list
    ::native_module_test_hdrs
    ::ref_cc
    iree::base::api
    iree::base::status
    iree::testing::gtest
    iree::testing::gtest_main
)

iree_cc_library(
  NAME
    native_module_test_hdrs
  HDRS
    "native_module_test.h"
  DEPS
    ::context
    ::instance
    ::native_module
    ::ref
    ::stack
    iree::base::api
  PUBLIC
)

iree_cc_library(
  NAME
    native_module_cc
  HDRS
    "module_abi_packing.h"
    "native_module_cc.h"
  DEPS
    ::builtin_types
    ::module
    ::ref
    ::ref_cc
    ::stack
    absl::inlined_vector
    absl::optional
    absl::span
    absl::strings
    iree::base::api
    iree::base::ref_ptr
    iree::base::status
  PUBLIC
)

iree_cc_library(
  NAME
    ref
  HDRS
    "ref.h"
  SRCS
    "ref.c"
  DEPS
    iree::base::api
    iree::base::atomics
  PUBLIC
)

iree_cc_test(
  NAME
    ref_test
  SRCS
    "ref_test.cc"
  DEPS
    ::ref
    iree::base::api
    iree::base::ref_ptr
    iree::testing::gtest
    iree::testing::gtest_main
)

iree_cc_library(
  NAME
    ref_cc
  HDRS
    "ref_cc.h"
  DEPS
    ::ref
    absl::core_headers
    iree::base::api
  PUBLIC
)

iree_cc_library(
  NAME
    stack
  HDRS
    "stack.h"
  SRCS
    "stack.c"
  DEPS
    ::module
    ::ref
    iree::base::alignment
    iree::base::api
    iree::base::tracing
  PUBLIC
)

iree_cc_test(
  NAME
    stack_test
  SRCS
    "stack_test.cc"
  DEPS
    ::ref
    ::stack
    iree::base::api
    iree::base::ref_ptr
    iree::testing::gtest
    iree::testing::gtest_main
)

iree_cc_library(
  NAME
    type_def
  HDRS
    "type_def.h"
  DEPS
    ::ref
    ::value
  PUBLIC
)

iree_cc_library(
  NAME
    value
  HDRS
    "value.h"
  PUBLIC
)

iree_cc_library(
  NAME
    vm
  HDRS
    "api.h"
  DEPS
    ::builtin_types
    ::context
    ::instance
    ::invocation
    ::list
    ::module
    ::native_module
    ::ref
    ::stack
    ::type_def
    ::value
    iree::base::api
  PUBLIC
)