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",
    ],
)