# 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

# WebGPU SPIR-V compiler target: compiles MLIR to SPIR-V, then translates
# SPIR-V to WGSL via Tint (from Dawn). Output is a FlatBuffer executable
# containing WGSL shader sources.

load("//build_tools/bazel:build_defs.oss.bzl", "iree_compiler_cc_library", "iree_compiler_register_plugin")

package(
    default_visibility = ["//visibility:public"],
    features = ["layering_check"],
    licenses = ["notice"],
)

iree_compiler_register_plugin(
    plugin_id = "hal_target_webgpu_spirv",
    target = ":WebGPUSPIRV",
)

iree_compiler_cc_library(
    name = "SPIRVToWGSL",
    srcs = [
        "SPIRVToWGSL.cpp",
    ],
    hdrs = [
        "SPIRVToWGSL.h",
    ],
    # Tint's public headers use C++20 features such as concepts and std::span.
    # Some Dawn headers use warning pragmas newer than CI clang versions.
    copts = [
        "-std=c++20",
        "-Wno-unknown-warning-option",
    ],
    deps = [
        "@dawn//src/tint/lang/core/ir",
        "@dawn//src/tint/lang/spirv/reader",
        "@dawn//src/tint/lang/wgsl/program",
        "@dawn//src/tint/lang/wgsl/writer",
        "@llvm-project//llvm:Support",
    ],
)

iree_compiler_cc_library(
    name = "WebGPUSPIRV",
    srcs = [
        "WebGPUSPIRVTarget.cpp",
    ],
    deps = [
        ":SPIRVToWGSL",
        "//compiler/src/iree/compiler/Codegen/Common:PassHeaders",
        "//compiler/src/iree/compiler/Codegen/Dialect/Codegen/IR:IREECodegenDialect",
        "//compiler/src/iree/compiler/Codegen/Dialect/GPU/TargetUtils:KnownTargets",
        "//compiler/src/iree/compiler/Codegen/SPIRV",
        "//compiler/src/iree/compiler/Codegen/Utils",
        "//compiler/src/iree/compiler/Codegen/WGSL",
        "//compiler/src/iree/compiler/Dialect/Flow/IR",
        "//compiler/src/iree/compiler/Dialect/HAL/Target",
        "//compiler/src/iree/compiler/Dialect/HAL/Utils:ExecutableDebugInfoUtils",
        "//compiler/src/iree/compiler/PluginAPI",
        "//compiler/src/iree/compiler/Utils",
        "//runtime/src/iree/schemas:executable_debug_info_c_fbs",
        "//runtime/src/iree/schemas:webgpu_executable_def_c_fbs",
        "@llvm-project//llvm:Support",
        "@llvm-project//mlir:GPUDialect",
        "@llvm-project//mlir:IR",
        "@llvm-project//mlir:SPIRVDialect",
        "@llvm-project//mlir:SPIRVSerialization",
        "@llvm-project//mlir:SPIRVTransforms",
        "@llvm-project//mlir:Support",
    ],
)