# Copyright 2026 The IREE Authors
#
# Licensed under the Apache License v2.0 with LLVM Exceptions.
# See https://llvm.org/LICENSE.txt for license information.
# SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception

load("//build_tools/bazel:iree_check_test.bzl", "iree_check_single_backend_test_suite")

package(
    features = ["layering_check"],
    licenses = ["notice"],  # Apache 2.0
)

iree_check_single_backend_test_suite(
    name = "check_llvm-cpu_local-task",
    srcs = [
        "contraction_scaling.mlir",
    ],
    compiler_flags = [
        "--iree-llvmcpu-target-cpu=generic",
        "--iree-global-opt-enable-qdq-to-integer-math=true",
    ],
    driver = "local-task",
    # Accepted numerical differences have separate reference/rewrite checks.
    runner_args = ["--gtest_filter=-*_reference"],
    target_backend = "llvm-cpu",
)

iree_check_single_backend_test_suite(
    name = "check_reference_llvm-cpu_local-task",
    srcs = [
        "contraction_scaling.mlir",
    ],
    compiler_flags = [
        "--iree-global-opt-enable-qdq-to-integer-math=false",
        "--iree-llvmcpu-target-cpu=generic",
    ],
    driver = "local-task",
    runner_args = ["--gtest_filter=-*_rewritten"],
    target_backend = "llvm-cpu",
)