| """Utilities for handling hand-written SPIR-V files.""" |
| |
| load("//:build_defs.google.bzl", "iree_glsl_vulkan") |
| load("///build_tools/embed_data:build_defs.bzl", "cc_embed_data") |
| |
| def spirv_kernel_cc_library(name, srcs): |
| """Compiles GLSL files into SPIR-V binaries and embeds them in a cc_library. |
| |
| Args: |
| name: cc_library name to depend on. |
| srcs: a list of GLSL source files. |
| """ |
| spv_files = [] |
| for src in srcs: |
| spv_name = src.split(".")[-2] |
| iree_glsl_vulkan( |
| name = spv_name, |
| srcs = [src], |
| ) |
| spv_files.append(spv_name + ".spv") |
| native.filegroup( |
| name = name + "_files", |
| srcs = spv_files, |
| ) |
| cc_embed_data( |
| name = name, |
| srcs = spv_files, |
| cc_file_output = name + ".cc", |
| h_file_output = name + ".h", |
| cpp_namespace = "mlir::iree_compiler::spirv_kernels", |
| flatten = True, |
| ) |
