| ################################################################################ |
| # Autogenerated by build_tools/bazel_to_cmake/bazel_to_cmake.py from # |
| # tests/e2e/models/BUILD # |
| # # |
| # Use iree_cmake_extra_content from iree/build_defs.oss.bzl to add arbitrary # |
| # CMake-only content. # |
| # # |
| # To disable autogeneration for this file entirely, delete this header. # |
| ################################################################################ |
| |
| iree_add_all_subdirs() |
| |
| iree_lit_test_suite( |
| NAME |
| lit |
| SRCS |
| "collatz.mlir" |
| "edge_detection.mlir" |
| "fragment_000.mlir" |
| "fullyconnected.mlir" |
| "mnist_fake_weights.mlir" |
| "resnet50_fake_weights.mlir" |
| "unidirectional_lstm.mlir" |
| TOOLS |
| ${IREE_LLD_TARGET} |
| FileCheck |
| iree-run-mlir |
| LABELS |
| "hostonly" |
| "optonly" |
| TIMEOUT |
| 900 |
| ) |
| |
| iree_check_single_backend_test_suite( |
| NAME |
| check_dylib-llvm-aot_dylib |
| SRCS |
| "bert_encoder_unrolled_fake_weights.mlir" |
| "mobilenetv3_fake_weights.mlir" |
| TARGET_BACKEND |
| "dylib-llvm-aot" |
| DRIVER |
| "local-task" |
| COMPILER_FLAGS |
| "--iree-input-type=mhlo" |
| ) |
| |
| iree_check_single_backend_test_suite( |
| NAME |
| check_vulkan-spirv_vulkan |
| SRCS |
| "bert_encoder_unrolled_fake_weights.mlir" |
| "mobilenetv3_fake_weights.mlir" |
| TARGET_BACKEND |
| "vulkan-spirv" |
| DRIVER |
| "vulkan" |
| COMPILER_FLAGS |
| "--iree-input-type=mhlo" |
| TIMEOUT |
| 900 |
| ) |
| |
| iree_check_single_backend_test_suite( |
| NAME |
| check_cuda_cuda |
| SRCS |
| "bert_encoder_unrolled_fake_weights.mlir" |
| "mobilenetv3_fake_weights.mlir" |
| TARGET_BACKEND |
| "cuda" |
| DRIVER |
| "cuda" |
| COMPILER_FLAGS |
| "--iree-input-type=mhlo" |
| LABELS |
| "noasan" |
| "nomsan" |
| "notsan" |
| "noubsan" |
| "requires-gpu-nvidia" |
| TIMEOUT |
| 900 |
| ) |
| |
| ### BAZEL_TO_CMAKE_PRESERVES_ALL_CONTENT_BELOW_THIS_LINE ### |
