onnx-mlir/src/BUILD

package(
    default_visibility = ["//visibility:public"],
)

genrule(
    name = "ExternalUtil",
    srcs = [
        "ExternalUtil.hpp.in",
    ],
    outs = [
        "ExternalUtil.hpp",
    ],
    cmd = "cp $(location :ExternalUtil.hpp.in) $(location :ExternalUtil.hpp)",
)

cc_library(
    name = "MainUtils",
    srcs = ["MainUtils.cpp"],
    hdrs = [
        "MainUtils.hpp",
        ":ExternalUtil",
    ],
    deps = [
        "//src/Builder:OMBuilder",
        "//src/Dialect/Krnl:OMKrnlOps",
        "//src/Dialect/ONNX:OMONNXOps",
        "//src/Pass:Passes",
        "@llvm-project//mlir:AllPassesAndDialectsNoRegistration",
        "@onnx",
    ],
)

cc_binary(
    name = "onnx-mlir",
    srcs = ["main.cpp"],
    copts = [],
    deps = [
        ":MainUtils",
        "//src/Conversion/KrnlToAffine:OMKrnlToAffine",
        "//src/Conversion/KrnlToLLVM:OMKrnlToLLVM",
        "//src/Conversion/ONNXToKrnl:OMONNXToKrnl",
        "//src/Transform:OMBundleMemoryPools",
        "//src/Transform:OMDisconnectKrnlDimFromAlloc",
        "//src/Transform:OMElideKrnlGlobalConstants",
        "//src/Transform:OMEnableMemoryPool",
        "//src/Transform:OMLowerKrnlShape",
        "//src/Transform:OMPackKrnlGlobalConstants",
        "//src/Transform/ONNX:OMAttributePromotion",
        "//src/Transform/ONNX:OMElideConstants",
        "//src/Transform/ONNX:OMONNXRewrite",
        "//src/Transform/ONNX:OMShapeInference",
    ],
)