onnx-mlir/src/Dialect/Krnl/BUILD

39 lines
836 B
Python

load("@llvm-project//mlir:tblgen.bzl", "gentbl")
gentbl(
name = "OMKrnlOpsIncGen",
tbl_outs = [
("-gen-op-decls", "KrnlOps.hpp.inc"),
("-gen-op-defs", "KrnlOps.cpp.inc"),
],
tblgen = "@llvm-project//mlir:mlir-tblgen",
td_file = "KrnlOps.td",
td_includes = ["."],
td_srcs = [
"@llvm-project//mlir:TdFiles",
],
)
cc_library(
name = "OMKrnlOps",
srcs = [
"KrnlHelper.cpp",
"KrnlHelper.hpp",
"KrnlOps.cpp",
"KrnlOps.hpp",
"KrnlTypes.cpp",
"KrnlTypes.hpp",
],
hdrs = glob([
"**/*.h",
"**/*.inc",
]),
deps = [
":OMKrnlOpsIncGen",
"@llvm-project//mlir:Affine",
"@llvm-project//mlir:IR",
"@llvm-project//mlir:Shape",
"@llvm-project//mlir:TableGen",
],
)