blob: c7e5e51c2ec306b2454f442d26a63ce41cd9bd17 [file]
# A collection of hand-written kernels for special ops that cannot (yet) be
# generated by the compiler.
#
# This is a grab bag right now during bringup with little structure. If you
# have ideas on how to make this better please ping us :)
#
# To use the kernels add a dep against this package and use
# mlir::iree_compiler::spirv::Kernels_create() to get the FileToc. Kernels
# will have filenames matching their original with the extension changed to
# '.spv' (for example 'add.comp' -> 'add.spv').
load(":spirv_utils.bzl", "spirv_kernel_cc_library")
package(
default_visibility = ["//visibility:public"],
licenses = ["notice"], # Apache 2.0
)
spirv_kernel_cc_library(
name = "Kernels",
srcs = [
"matmul.comp",
"reduce_untiled.comp",
],
)