################################################################################
# Autogenerated by build_tools/bazel_to_cmake/bazel_to_cmake.py from           #
# tests/transform_dialect/cuda/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()

if(NOT IREE_HAL_DRIVER_CUDA OR NOT IREE_TARGET_BACKEND_CUDA)
  return()
endif()

iree_lit_test_suite(
  NAME
    lit
  SRCS
    "mma.mlir"
    "reduction.mlir"
    "reduction_eltwise.mlir"
    "reduction_v2.mlir"
    "reduction_v2_uneven.mlir"
    "softmax.mlir"
    "softmax_partial.mlir"
    "softmax_v2.mlir"
  TOOLS
    FileCheck
    iree-compile
    iree-opt
    iree-run-module
  DATA
    mma_reduction_layout_analysis_codegen_spec.mlir
    mma_reduction_layout_analysis_dispatch_spec.mlir
    mma_using_layout_analysis_codegen_spec.mlir
    reduction_codegen_spec.mlir
    reduction_eltwise_codegen_spec.mlir
    reduction_v2_codegen_spec.mlir
    softmax_codegen_spec.mlir
    softmax_dispatch_spec.mlir
    softmax_partial_codegen_spec.mlir
    softmax_v2_codegen_spec.mlir
  LABELS
    "noasan"
    "nomsan"
    "notsan"
    "noubsan"
    "requires-gpu-nvidia"
    "driver=cuda"
)

iree_lit_test_suite(
  NAME
    sm80_lit
  SRCS
    "double_mma_layout_analysis.mlir"
    "mma_elemwise_layout_analysis.mlir"
    "mma_reduction_layout_analysis.mlir"
    "mma_using_layout_analysis.mlir"
  TOOLS
    FileCheck
    iree-compile
    iree-opt
    iree-run-module
  DATA
    double_mma_layout_analysis_codegen_spec.mlir
    double_mma_layout_analysis_dispatch_spec.mlir
    mma_elemwise_layout_analysis_codegen_spec.mlir
    mma_reduction_layout_analysis_codegen_spec.mlir
    mma_reduction_layout_analysis_dispatch_spec.mlir
    mma_using_layout_analysis_codegen_spec.mlir
  LABELS
    "noasan"
    "nomsan"
    "notsan"
    "noubsan"
    "requires-gpu-sm80"
    "driver=cuda"
)

### BAZEL_TO_CMAKE_PRESERVES_ALL_CONTENT_BELOW_THIS_LINE ###
