77 lines
2.1 KiB
Python
77 lines
2.1 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",
|
|
"@llvm-project//mlir:TdFiles",
|
|
],
|
|
)
|
|
|
|
cc_library(
|
|
name = "OMONNXOps",
|
|
srcs = [
|
|
"ONNXOps.cpp",
|
|
"ONNXOpsHelper.cpp",
|
|
],
|
|
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:AffineOps",
|
|
"@llvm-project//mlir:Dialect",
|
|
"@llvm-project//mlir:ExecutionEngine",
|
|
"@llvm-project//mlir:LLVMDialect",
|
|
# "@llvm-project//mlir:LoopOps",
|
|
"@llvm-project//mlir:MlirOptLib",
|
|
"@llvm-project//mlir:Parser",
|
|
"@llvm-project//mlir:StandardOps",
|
|
"@llvm-project//mlir:Support",
|
|
"@llvm-project//mlir:VectorOps",
|
|
|
|
# "@llvm-project//mlir:Support",
|
|
# "@llvm-project//mlir:TableGen",
|
|
],
|
|
)
|
|
|
|
# "@llvm-project//llvm:core",
|
|
# "@llvm-project//llvm:dtu_asm_parser",
|
|
# "@llvm-project//llvm:dtu_code_gen",
|
|
# "@llvm-project//llvm:dtu_desc",
|
|
# "@llvm-project//llvm:orc_jit",
|
|
# "@llvm-project//llvm:passes",
|
|
# "@llvm-project//llvm:x86_asm_parser",
|
|
# "@llvm-project//llvm:x86_code_gen",
|
|
# "@llvm-project//mlir:AffineOps",
|
|
# "@llvm-project//mlir:Dialect",
|
|
# "@llvm-project//mlir:ExecutionEngine",
|
|
# "@llvm-project//mlir:LLVMDialect",
|
|
# "@llvm-project//mlir:LoopOps",
|
|
# "@llvm-project//mlir:MlirOptLib",
|
|
# "@llvm-project//mlir:Parser",
|
|
# "@llvm-project//mlir:StandardOps",
|
|
# "@llvm-project//mlir:Support",
|
|
# "@llvm-project//mlir:VectorOps",
|