blob: 17f1a5d654be9e64c1d02bdaafb0a64d85e70ae1 [file] [log] [blame]
################################################################################
# Autogenerated by build_tools/bazel_to_cmake/bazel_to_cmake.py from #
# tests/e2e/linalg/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
"conv2d.mlir"
"f32_to_i4.mlir"
"i4_to_f32.mlir"
TARGET_BACKEND
"llvm-cpu"
DRIVER
"local-task"
)
iree_check_single_backend_test_suite(
NAME
check_vmvx_local-task
SRCS
"conv2d.mlir"
TARGET_BACKEND
"vmvx"
DRIVER
"local-task"
)
iree_check_single_backend_test_suite(
NAME
check_vulkan-spirv_vulkan
SRCS
"conv2d.mlir"
"i4_to_f32.mlir"
TARGET_BACKEND
"vulkan-spirv"
DRIVER
"vulkan"
)
iree_check_single_backend_test_suite(
NAME
check_winograd_llvm-cpu_local-task
SRCS
"conv2d.mlir"
TARGET_BACKEND
"llvm-cpu"
DRIVER
"local-task"
COMPILER_FLAGS
"--iree-preprocessing-pass-pipeline=builtin.module\(func.func\(iree-linalg-ext-convert-conv2d-to-winograd\)\)"
)
iree_check_single_backend_test_suite(
NAME
check_winograd_vulkan-spirv_vulkan
SRCS
"conv2d.mlir"
TARGET_BACKEND
"vulkan-spirv"
DRIVER
"vulkan"
COMPILER_FLAGS
"--iree-preprocessing-pass-pipeline=builtin.module\(func.func\(iree-linalg-ext-convert-conv2d-to-winograd\)\)"
)
iree_check_single_backend_test_suite(
NAME
check_large_linalg_matmul_cuda
SRCS
"conv2d.mlir"
"f32_to_i4.mlir"
"i4_to_f32.mlir"
"large_linalg_matmul.mlir"
TARGET_BACKEND
"cuda"
DRIVER
"cuda"
LABELS
"noasan"
"nomsan"
"notsan"
"noubsan"
"requires-gpu-nvidia"
)
### BAZEL_TO_CMAKE_PRESERVES_ALL_CONTENT_BELOW_THIS_LINE ###