56 lines
1.3 KiB
Python
56 lines
1.3 KiB
Python
|
package(
|
||
|
|
default_visibility = ["//visibility:public"],
|
||
|
|
)
|
||
|
|
|
||
|
|
load("@llvm-project//mlir:tblgen.bzl", "gentbl")
|
||
|
|
|
||
|
|
gentbl(
|
||
|
|
name = "OMONNXOpsIncGen",
|
||
|
|
tbl_outs = [
|
||
|
|
("-gen-op-decls", "ONNXOps.hpp.inc"),
|
||
|
|
("-gen-op-defs", "ONNXOps.cpp.inc"),
|
||
|
|
],
|
||
|
|
tblgen = "@llvm-project//mlir:mlir-tblgen",
|
||
|
|
td_file = "ONNXOps.td",
|
||
|
|
td_includes = ["."],
|
||
|
|
td_srcs = [
|
||
|
|
"ONNXOps.td.inc",
|
||
|
|
# "//src/Interface:PromotableConstOperandsOpInterface.td",
|
||
|
|
# "//src/Interface:ResultTypeInferenceOpInterface.td",
|
||
|
|
# "//src/Interface:ShapeInferenceInterface.td",
|
||
|
|
|
||
|
|
"@llvm-project//mlir:TdFiles",
|
||
|
|
],
|
||
|
|
)
|
||
|
|
|
||
|
|
cc_library(
|
||
|
|
name = "OMONNXOps",
|
||
|
|
srcs = [
|
||
|
|
"ONNXOps.cpp",
|
||
|
|
# "ONNXOps.hpp",
|
||
|
|
"ONNXOpsHelper.cpp",
|
||
|
|
# "ONNXOpsHelper.hpp",
|
||
|
|
],
|
||
|
|
hdrs = glob([
|
||
|
|
"**/*.hpp",
|
||
|
|
"**/*.inc",
|
||
|
|
]),
|
||
|
|
deps = [
|
||
|
|
":OMONNXOpsIncGen",
|
||
|
|
|
||
|
|
"@//src/Interface:OMPromotableConstOperandsOpInterface",
|
||
|
|
"@//src/Interface:OMShapeInferenceOpInterface",
|
||
|
|
"@//src/Interface:OMResultTypeInferenceOpInterface",
|
||
|
|
|
||
|
|
"@onnx",
|
||
|
|
# "@onnx//:onnx_proto_genproto",
|
||
|
|
# "@llvm-project//mlir:Affine",
|
||
|
|
"@llvm-project//mlir:IR",
|
||
|
|
"@llvm-project//mlir:Shape",
|
||
|
|
|
||
|
|
# "@llvm-project//mlir:Support",
|
||
|
|
# "@llvm-project//mlir:TableGen",
|
||
|
|
|
||
|
|
],
|
||
|
|
)
|