tests/e2e/stablehlo_ops/CMakeLists.txt

################################################################################
# Autogenerated by build_tools/bazel_to_cmake/bazel_to_cmake.py from           #
# tests/e2e/stablehlo_ops/BUILD.bazel                                          #
#                                                                              #
# 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_check_single_backend_test_suite(
  NAME
    check_llvm-cpu_local-task
  SRCS
    "abs.mlir"
    "add.mlir"
    "batch_norm_inference.mlir"
    "bitcast_convert.mlir"
    "broadcast.mlir"
    "broadcast_add.mlir"
    "broadcast_in_dim.mlir"
    "clamp.mlir"
    "compare.mlir"
    "complex.mlir"
    "concatenate.mlir"
    "constant.mlir"
    "convert.mlir"
    "convolution.mlir"
    "cosine.mlir"
    "divide.mlir"
    "dot.mlir"
    "dot_bf16.mlir"
    "dot_general.mlir"
    "dynamic_slice.mlir"
    "dynamic_update_slice.mlir"
    "exponential.mlir"
    "exponential_fp16.mlir"
    "exponential_minus_one.mlir"
    "fft.mlir"
    "finite.mlir"
    "floor.mlir"
    "gather.mlir"
    "iota.mlir"
    "log.mlir"
    "log_plus_one.mlir"
    "maximum.mlir"
    "minimum.mlir"
    "multiply.mlir"
    "negate.mlir"
    "pad.mlir"
    "philox.mlir"
    "pow.mlir"
    "reduce.mlir"
    "reduce_window.mlir"
    "remainder.mlir"
    "reshape.mlir"
    "reverse.mlir"
    "rng_normal.mlir"
    "rng_uniform.mlir"
    "round.mlir"
    "rsqrt.mlir"
    "scatter.mlir"
    "scatter_dynamic.mlir"
    "select.mlir"
    "sine.mlir"
    "slice.mlir"
    "sort.mlir"
    "sqrt.mlir"
    "subtract.mlir"
    "tanh.mlir"
    "three_fry.mlir"
    "torch_index_select.mlir"
    "transpose.mlir"
    "while.mlir"
  TARGET_BACKEND
    "llvm-cpu"
  DRIVER
    "local-task"
  COMPILER_FLAGS
    "--iree-input-type=stablehlo"
)

iree_check_single_backend_test_suite(
  NAME
    check_vmvx_local-task
  SRCS
    "abs.mlir"
    "add.mlir"
    "batch_norm_inference.mlir"
    "bitcast_convert.mlir"
    "broadcast.mlir"
    "broadcast_add.mlir"
    "broadcast_in_dim.mlir"
    "clamp.mlir"
    "compare.mlir"
    "complex.mlir"
    "concatenate.mlir"
    "constant.mlir"
    "convert.mlir"
    "convolution.mlir"
    "cosine.mlir"
    "divide.mlir"
    "dot.mlir"
    "dot_general.mlir"
    "dynamic_slice.mlir"
    "dynamic_update_slice.mlir"
    "exponential.mlir"
    "exponential_minus_one.mlir"
    "fft.mlir"
    "finite.mlir"
    "floor.mlir"
    "gather.mlir"
    "iota.mlir"
    "log.mlir"
    "log_plus_one.mlir"
    "maximum.mlir"
    "minimum.mlir"
    "multiply.mlir"
    "negate.mlir"
    "pad.mlir"
    "philox.mlir"
    "pow.mlir"
    "reduce.mlir"
    "reduce_window.mlir"
    "remainder.mlir"
    "reshape.mlir"
    "reverse.mlir"
    "rng_normal.mlir"
    "rng_uniform.mlir"
    "round.mlir"
    "rsqrt.mlir"
    "scatter.mlir"
    "scatter_dynamic.mlir"
    "select.mlir"
    "sine.mlir"
    "slice.mlir"
    "sort.mlir"
    "sqrt.mlir"
    "subtract.mlir"
    "tanh.mlir"
    "three_fry.mlir"
    "torch_index_select.mlir"
    "transpose.mlir"
    "while.mlir"
  TARGET_BACKEND
    "vmvx"
  DRIVER
    "local-task"
  COMPILER_FLAGS
    "--iree-input-type=stablehlo"
)

iree_check_single_backend_test_suite(
  NAME
    check_vulkan-spirv_vulkan
  SRCS
    "abs.mlir"
    "add.mlir"
    "batch_norm_inference.mlir"
    "bitcast_convert.mlir"
    "broadcast.mlir"
    "broadcast_add.mlir"
    "broadcast_in_dim.mlir"
    "clamp.mlir"
    "compare.mlir"
    "complex.mlir"
    "concatenate.mlir"
    "constant.mlir"
    "convert.mlir"
    "convolution.mlir"
    "cosine.mlir"
    "divide.mlir"
    "dot.mlir"
    "dot_bf16.mlir"
    "dot_general.mlir"
    "dynamic_slice.mlir"
    "dynamic_update_slice.mlir"
    "exponential.mlir"
    "exponential_minus_one.mlir"
    "finite.mlir"
    "floor.mlir"
    "gather.mlir"
    "iota.mlir"
    "log.mlir"
    "log_plus_one.mlir"
    "maximum.mlir"
    "minimum.mlir"
    "multiply.mlir"
    "negate.mlir"
    "pad.mlir"
    "philox.mlir"
    "pow.mlir"
    "reduce.mlir"
    "reduce_window.mlir"
    "remainder.mlir"
    "reshape.mlir"
    "rng_normal.mlir"
    "rng_uniform.mlir"
    "round.mlir"
    "rsqrt.mlir"
    "scatter.mlir"
    "scatter_dynamic.mlir"
    "select.mlir"
    "sine.mlir"
    "slice.mlir"
    "sort.mlir"
    "sqrt.mlir"
    "subtract.mlir"
    "tanh.mlir"
    "three_fry.mlir"
    "torch_index_select.mlir"
    "transpose.mlir"
    "while.mlir"
  TARGET_BACKEND
    "vulkan-spirv"
  DRIVER
    "vulkan"
  COMPILER_FLAGS
    "--iree-input-type=stablehlo"
)

iree_check_single_backend_test_suite(
  NAME
    check_llvm-cpu-host_local-task
  SRCS
    "abs.mlir"
    "add.mlir"
    "batch_norm_inference.mlir"
    "bitcast_convert.mlir"
    "broadcast.mlir"
    "broadcast_add.mlir"
    "broadcast_in_dim.mlir"
    "clamp.mlir"
    "compare.mlir"
    "complex.mlir"
    "concatenate.mlir"
    "constant.mlir"
    "convert.mlir"
    "convolution.mlir"
    "cosine.mlir"
    "divide.mlir"
    "dot.mlir"
    "dot_bf16.mlir"
    "dot_general.mlir"
    "dynamic_slice.mlir"
    "dynamic_update_slice.mlir"
    "exponential.mlir"
    "exponential_fp16.mlir"
    "exponential_minus_one.mlir"
    "fft.mlir"
    "finite.mlir"
    "floor.mlir"
    "gather.mlir"
    "iota.mlir"
    "log.mlir"
    "log_plus_one.mlir"
    "maximum.mlir"
    "minimum.mlir"
    "multiply.mlir"
    "negate.mlir"
    "pad.mlir"
    "philox.mlir"
    "pow.mlir"
    "reduce.mlir"
    "reduce_window.mlir"
    "remainder.mlir"
    "reshape.mlir"
    "reverse.mlir"
    "rng_normal.mlir"
    "rng_uniform.mlir"
    "round.mlir"
    "rsqrt.mlir"
    "scatter.mlir"
    "scatter_dynamic.mlir"
    "select.mlir"
    "sine.mlir"
    "slice.mlir"
    "sort.mlir"
    "sqrt.mlir"
    "subtract.mlir"
    "tanh.mlir"
    "three_fry.mlir"
    "torch_index_select.mlir"
    "transpose.mlir"
    "while.mlir"
  TARGET_BACKEND
    "llvm-cpu"
  DRIVER
    "local-task"
  COMPILER_FLAGS
    "--iree-input-type=stablehlo"
    "--iree-llvmcpu-target-cpu-features=host"
  LABELS
    "hostonly"
    "local"
)

iree_check_single_backend_test_suite(
  NAME
    check_cuda_graph
  SRCS
    "abs.mlir"
    "add.mlir"
    "batch_norm_inference.mlir"
    "bitcast_convert.mlir"
    "broadcast.mlir"
    "broadcast_add.mlir"
    "broadcast_in_dim.mlir"
    "clamp.mlir"
    "compare.mlir"
    "complex.mlir"
    "concatenate.mlir"
    "constant.mlir"
    "convert.mlir"
    "convolution.mlir"
    "cosine.mlir"
    "divide.mlir"
    "dot.mlir"
    "dot_bf16.mlir"
    "dot_general.mlir"
    "dynamic_slice.mlir"
    "dynamic_update_slice.mlir"
    "exponential.mlir"
    "exponential_fp16.mlir"
    "exponential_minus_one.mlir"
    "fft.mlir"
    "finite.mlir"
    "floor.mlir"
    "gather.mlir"
    "iota.mlir"
    "log.mlir"
    "log_plus_one.mlir"
    "maximum.mlir"
    "minimum.mlir"
    "multiply.mlir"
    "negate.mlir"
    "pad.mlir"
    "philox.mlir"
    "pow.mlir"
    "reduce.mlir"
    "reduce_window.mlir"
    "remainder.mlir"
    "reshape.mlir"
    "reverse.mlir"
    "rng_normal.mlir"
    "rng_uniform.mlir"
    "round.mlir"
    "rsqrt.mlir"
    "scatter.mlir"
    "scatter_dynamic.mlir"
    "select.mlir"
    "sine.mlir"
    "slice.mlir"
    "sort.mlir"
    "sqrt.mlir"
    "subtract.mlir"
    "tanh.mlir"
    "three_fry.mlir"
    "torch_index_select.mlir"
    "transpose.mlir"
    "while.mlir"
  TARGET_BACKEND
    "cuda"
  DRIVER
    "cuda"
  COMPILER_FLAGS
    "--iree-input-type=stablehlo"
    "--iree-stream-emulate-memset"
  RUNNER_ARGS
    "--cuda_use_streams=false"
  LABELS
    "noasan"
    "nomsan"
    "notsan"
    "noubsan"
    "requires-gpu-nvidia"
)

iree_check_single_backend_test_suite(
  NAME
    check_cuda_streams
  SRCS
    "abs.mlir"
    "add.mlir"
    "batch_norm_inference.mlir"
    "bitcast_convert.mlir"
    "broadcast.mlir"
    "broadcast_add.mlir"
    "broadcast_in_dim.mlir"
    "clamp.mlir"
    "compare.mlir"
    "complex.mlir"
    "concatenate.mlir"
    "constant.mlir"
    "convert.mlir"
    "convolution.mlir"
    "cosine.mlir"
    "divide.mlir"
    "dot.mlir"
    "dot_bf16.mlir"
    "dot_general.mlir"
    "dynamic_slice.mlir"
    "dynamic_update_slice.mlir"
    "exponential.mlir"
    "exponential_fp16.mlir"
    "exponential_minus_one.mlir"
    "fft.mlir"
    "finite.mlir"
    "floor.mlir"
    "gather.mlir"
    "iota.mlir"
    "log.mlir"
    "log_plus_one.mlir"
    "maximum.mlir"
    "minimum.mlir"
    "multiply.mlir"
    "negate.mlir"
    "pad.mlir"
    "philox.mlir"
    "pow.mlir"
    "reduce.mlir"
    "reduce_window.mlir"
    "remainder.mlir"
    "reshape.mlir"
    "reverse.mlir"
    "rng_normal.mlir"
    "rng_uniform.mlir"
    "round.mlir"
    "rsqrt.mlir"
    "scatter.mlir"
    "scatter_dynamic.mlir"
    "select.mlir"
    "sine.mlir"
    "slice.mlir"
    "sort.mlir"
    "sqrt.mlir"
    "subtract.mlir"
    "tanh.mlir"
    "three_fry.mlir"
    "torch_index_select.mlir"
    "transpose.mlir"
    "while.mlir"
  TARGET_BACKEND
    "cuda"
  DRIVER
    "cuda"
  COMPILER_FLAGS
    "--iree-input-type=stablehlo"
  RUNNER_ARGS
    "--cuda_use_streams=true"
  LABELS
    "noasan"
    "nomsan"
    "notsan"
    "noubsan"
    "requires-gpu-nvidia"
)

### BAZEL_TO_CMAKE_PRESERVES_ALL_CONTENT_BELOW_THIS_LINE ###

iree_check_single_backend_test_suite(
  NAME
    check_webgpu
  SRCS
    "abs.mlir"
    "add.mlir"
    "batch_norm_inference.mlir"
    "bitcast_convert.mlir"
    "broadcast.mlir"
    "broadcast_add.mlir"
    "broadcast_in_dim.mlir"
    # "clamp.mlir"  # TODO(#10906): fix (i8/i16?)
    # "compare.mlir"  # TODO(#10906): fix (i8/i16?)
    # "complex.mlir" # TODO(#11054)
    "concatenate.mlir"
    "constant.mlir"
    # "convert.mlir"  # TODO(#10906): fix (i8/i16?)
    "convolution.mlir"
    "cosine.mlir"
    "divide.mlir"
    "dot.mlir"
    "dot_general.mlir"
    "dynamic_slice.mlir"
    "dynamic_update_slice.mlir"
    "exponential.mlir"
    "exponential_fp16.mlir"
    "exponential_minus_one.mlir"
    # "fft.mlir"  # TODO(#9583): fix (fft codegen via spirv)
    # "finite.mlir"  # TODO(#11321): error: value cannot be represented as 'f32': inf
    "floor.mlir"
    "gather.mlir"
    "iota.mlir"
    "log.mlir"
    "log_plus_one.mlir"
    # "maximum.mlir"  # TODO(#10906): fix (i8/i16?)
    # "minimum.mlir"  # TODO(#10906): fix (i8/i16?)
    "multiply.mlir"
    "negate.mlir"
    "pad.mlir"
    # "philox.mlir"  # TODO(#12509): WebGPU SPIR-V broken
    "pow.mlir"
    "reduce.mlir"
    "reduce_window.mlir"
    "remainder.mlir"
    "reshape.mlir"
    "reverse.mlir"
    "rng_normal.mlir"
    "rng_uniform.mlir"
    "round.mlir"
    "rsqrt.mlir"
    "scatter.mlir"
    "scatter_dynamic.mlir"
    "select.mlir"
    "sine.mlir"
    "slice.mlir"
    "sort.mlir"
    "sqrt.mlir"
    "subtract.mlir"
    "tanh.mlir"
    # "three_fry.mlir"  # TODO(#12509): WebGPU SPIR-V broken
    "torch_index_select.mlir"
    "transpose.mlir"
    # "while.mlir"  # TODO(#12509): WebGPU SPIR-V broken
  TARGET_BACKEND
    "webgpu"
  # Only test compilation for now, the WebGPU driver is not stable/tested yet.
  # DRIVER
  #   "webgpu"
  COMPILER_FLAGS
    "--iree-input-type=stablehlo"
    "--iree-codegen-gpu-native-math-precision=true"  # TODO(#11321): Infer/flip default
)

# TODO(#14814): Re-enable once flaky segfaults are fixed.
# iree_check_single_backend_test_suite(
#   NAME
#     check_metal-spirv_metal
#   SRCS
#     "abs.mlir"
#     "add.mlir"
#     "batch_norm_inference.mlir"
#     "bitcast_convert.mlir"
#     "broadcast.mlir"
#     "broadcast_add.mlir"
#     "broadcast_in_dim.mlir"
#     "clamp.mlir"
#     "compare.mlir"
#     "complex.mlir"
#     "concatenate.mlir"
#     "constant.mlir"
#     "convert.mlir"
#     "convolution.mlir"
#     "cosine.mlir"
#     "divide.mlir"
#     "dot.mlir"
#     "dot_bf16.mlir"
#     "dot_general.mlir"
#     "dynamic_slice.mlir"
#     "dynamic_update_slice.mlir"
#     "exponential.mlir"
#     "exponential_fp16.mlir"
#     "exponential_minus_one.mlir"
#     "fft.mlir"
#     "finite.mlir"
#     "floor.mlir"
#     "gather.mlir"
#     "iota.mlir"
#     "log.mlir"
#     "log_plus_one.mlir"
#     "maximum.mlir"
#     "minimum.mlir"
#     "multiply.mlir"
#     "negate.mlir"
#     "pad.mlir"
#     "philox.mlir"
#     "pow.mlir"
#     "reduce.mlir"
#     # "reduce_window.mlir"  # TODO(15012): fix test crash
#     "remainder.mlir"
#     "reshape.mlir"
#     "reverse.mlir"
#     "rng_normal.mlir"
#     "rng_uniform.mlir"
#     "round.mlir"
#     "rsqrt.mlir"
#     "scatter.mlir"
#     "scatter_dynamic.mlir"
#     "select.mlir"
#     "sine.mlir"
#     "slice.mlir"
#     "sort.mlir"
#     "sqrt.mlir"
#     "subtract.mlir"
#     "tanh.mlir"
#     "three_fry.mlir"
#     "torch_index_select.mlir"
#     "transpose.mlir"
#     "while.mlir"
#   TARGET_BACKEND
#     "metal-spirv"
#   DRIVER
#     "metal"
#   COMPILER_FLAGS
#     "--iree-input-type=stablehlo"
# )