colin
/
mlir-hlo
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity
mlir-hlo/BUILD

1135 lines
33 KiB
Python
Raw Normal View History

Add Bazel build configuration to MLIR HLO standalone repo. Tested that this works in OSS in tree with `bazel build //...` and as a submodule in IREE. PiperOrigin-RevId: 347919106
2020-12-17 08:22:14 +08:00
load("@llvm-project//mlir:tblgen.bzl", "gentbl")
package(
default_visibility = ["//visibility:public"],
licenses = ["notice"], # Apache 2.0
)
exports_files(["include/mlir-hlo/Dialect/mhlo/IR/hlo_ops.td"])
exports_files(["include/mlir-hlo/Dialect/mhlo/IR/lhlo_ops.td"])
filegroup(
name = "hlo_ops_td_files",
srcs = [
"include/mlir-hlo/Dialect/mhlo/IR/chlo_ops.td",
"include/mlir-hlo/Dialect/mhlo/IR/hlo_ops.td",
"include/mlir-hlo/Dialect/mhlo/IR/hlo_ops_base.td",
"include/mlir-hlo/Dialect/mhlo/IR/hlo_ops_base_enums.td",
"include/mlir-hlo/Dialect/mhlo/IR/hlo_ops_base_structs.td",
"include/mlir-hlo/Dialect/mhlo/IR/hlo_utils.td",
"include/mlir-hlo/Dialect/mhlo/IR/infer_fusibility_op_interface.td",
"include/mlir-hlo/Dialect/mhlo/IR/lhlo_ops.td",
"include/mlir-hlo/Dialect/mhlo/IR/lhlo_ops_base.td",
"@llvm-project//mlir:OpBaseTdFiles",
"@llvm-project//mlir:SideEffectTdFiles",
"@llvm-project//mlir:include/mlir/Interfaces/CopyOpInterface.td",
"@llvm-project//mlir:include/mlir/Interfaces/InferTypeOpInterface.td",
"@llvm-project//mlir:include/mlir/Interfaces/LoopLikeInterface.td",
"@llvm-project//mlir:include/mlir/Interfaces/ViewLikeInterface.td",
],
)
filegroup(
name = "hlo_ops_base_td",
srcs = [
"include/mlir-hlo/Dialect/mhlo/IR/hlo_ops_base.td",
],
)
gentbl(
name = "MhloPassIncGen",
strip_include_prefix = "include",
tbl_outs = [
(
"-gen-pass-decls -name MHLO",
"include/mlir-hlo/Dialect/mhlo/transforms/mhlo_passes.h.inc",
),
],
tblgen = "@llvm-project//mlir:mlir-tblgen",
td_file = "include/mlir-hlo/Dialect/mhlo/transforms/mhlo_passes.td",
td_includes = [
Change hardcoded external/ include to match repository name This is the preferred name for the Bazel repository when included from another project. Note: These hardcoded "external/" includes are a horrible hack that we can hopefully get rid of soon with more robust Bazel tablegen rules. PiperOrigin-RevId: 350810768
2021-01-09 04:06:39 +08:00
"external/mlir-hlo/include",
Add Bazel build configuration to MLIR HLO standalone repo. Tested that this works in OSS in tree with `bazel build //...` and as a submodule in IREE. PiperOrigin-RevId: 347919106
2020-12-17 08:22:14 +08:00
"include",
],
td_srcs = [
"@llvm-project//mlir:PassBaseTdFiles",
],
)
gentbl(
name = "LmhloPassIncGen",
strip_include_prefix = "include",
tbl_outs = [
(
"-gen-pass-decls -name LMHLO",
"include/mlir-hlo/Dialect/mhlo/transforms/lmhlo_passes.h.inc",
),
],
tblgen = "@llvm-project//mlir:mlir-tblgen",
td_file = "include/mlir-hlo/Dialect/mhlo/transforms/lmhlo_passes.td",
td_includes = [
Change hardcoded external/ include to match repository name This is the preferred name for the Bazel repository when included from another project. Note: These hardcoded "external/" includes are a horrible hack that we can hopefully get rid of soon with more robust Bazel tablegen rules. PiperOrigin-RevId: 350810768
2021-01-09 04:06:39 +08:00
"external/mlir-hlo/include",
Add Bazel build configuration to MLIR HLO standalone repo. Tested that this works in OSS in tree with `bazel build //...` and as a submodule in IREE. PiperOrigin-RevId: 347919106
2020-12-17 08:22:14 +08:00
"include",
],
td_srcs = [
"@llvm-project//mlir:PassBaseTdFiles",
],
)
gentbl(
name = "chlo_ops_inc_gen",
strip_include_prefix = "include",
tbl_outs = [
("-gen-op-decls", "include/mlir-hlo/Dialect/mhlo/IR/chlo_ops.h.inc"),
("-gen-op-defs", "include/mlir-hlo/Dialect/mhlo/IR/chlo_ops.cc.inc"),
],
tblgen = "@llvm-project//mlir:mlir-tblgen",
td_file = "include/mlir-hlo/Dialect/mhlo/IR/chlo_ops.td",
td_includes = [
Change hardcoded external/ include to match repository name This is the preferred name for the Bazel repository when included from another project. Note: These hardcoded "external/" includes are a horrible hack that we can hopefully get rid of soon with more robust Bazel tablegen rules. PiperOrigin-RevId: 350810768
2021-01-09 04:06:39 +08:00
"external/mlir-hlo/include",
Add Bazel build configuration to MLIR HLO standalone repo. Tested that this works in OSS in tree with `bazel build //...` and as a submodule in IREE. PiperOrigin-RevId: 347919106
2020-12-17 08:22:14 +08:00
"include",
],
td_relative_includes = [
"include",
],
td_srcs = [":hlo_ops_td_files"],
)
gentbl(
name = "hlo_ops_inc_gen",
strip_include_prefix = "include",
tbl_outs = [
("-gen-op-decls", "include/mlir-hlo/Dialect/mhlo/IR/hlo_ops.h.inc"),
("-gen-op-defs", "include/mlir-hlo/Dialect/mhlo/IR/hlo_ops.cc.inc"),
],
tblgen = "@llvm-project//mlir:mlir-tblgen",
td_file = "include/mlir-hlo/Dialect/mhlo/IR/hlo_ops.td",
td_includes = [
Change hardcoded external/ include to match repository name This is the preferred name for the Bazel repository when included from another project. Note: These hardcoded "external/" includes are a horrible hack that we can hopefully get rid of soon with more robust Bazel tablegen rules. PiperOrigin-RevId: 350810768
2021-01-09 04:06:39 +08:00
"external/mlir-hlo/include",
Add Bazel build configuration to MLIR HLO standalone repo. Tested that this works in OSS in tree with `bazel build //...` and as a submodule in IREE. PiperOrigin-RevId: 347919106
2020-12-17 08:22:14 +08:00
"include",
"include/mlir-hlo/Dialect/mhlo/IR/hlo_utils.td",
],
td_relative_includes = [
"include",
],
td_srcs = [
":hlo_ops_base_td",
":hlo_ops_td_files",
"@llvm-project//mlir:include/mlir/Interfaces/InferTypeOpInterface.td",
],
)
gentbl(
name = "hlo_ops_base_inc_gen",
strip_include_prefix = "include",
tbl_outs = [
("-gen-op-decls", "include/mlir-hlo/Dialect/mhlo/IR/hlo_ops_base.h.inc"),
("-gen-op-defs", "include/mlir-hlo/Dialect/mhlo/IR/hlo_ops_base.cc.inc"),
],
tblgen = "@llvm-project//mlir:mlir-tblgen",
td_file = "include/mlir-hlo/Dialect/mhlo/IR/hlo_ops_base.td",
td_includes = [
Change hardcoded external/ include to match repository name This is the preferred name for the Bazel repository when included from another project. Note: These hardcoded "external/" includes are a horrible hack that we can hopefully get rid of soon with more robust Bazel tablegen rules. PiperOrigin-RevId: 350810768
2021-01-09 04:06:39 +08:00
"external/mlir-hlo/include",
Add Bazel build configuration to MLIR HLO standalone repo. Tested that this works in OSS in tree with `bazel build //...` and as a submodule in IREE. PiperOrigin-RevId: 347919106
2020-12-17 08:22:14 +08:00
"include",
],
td_relative_includes = [
"include",
],
td_srcs = [":hlo_ops_td_files"],
)
gentbl(
name = "hlo_ops_base_structs_inc_gen",
tbl_outs = [
("-gen-struct-attr-decls", "include/mlir-hlo/Dialect/mhlo/IR/hlo_ops_base_structs.h.inc"),
("-gen-struct-attr-defs", "include/mlir-hlo/Dialect/mhlo/IR/hlo_ops_base_structs.cc.inc"),
],
tblgen = "@llvm-project//mlir:mlir-tblgen",
td_file = "include/mlir-hlo/Dialect/mhlo/IR/hlo_ops_base.td",
td_includes = [
Change hardcoded external/ include to match repository name This is the preferred name for the Bazel repository when included from another project. Note: These hardcoded "external/" includes are a horrible hack that we can hopefully get rid of soon with more robust Bazel tablegen rules. PiperOrigin-RevId: 350810768
2021-01-09 04:06:39 +08:00
"external/mlir-hlo/include",
Add Bazel build configuration to MLIR HLO standalone repo. Tested that this works in OSS in tree with `bazel build //...` and as a submodule in IREE. PiperOrigin-RevId: 347919106
2020-12-17 08:22:14 +08:00
"include",
],
td_relative_includes = [
"include",
],
td_srcs = [":hlo_ops_td_files"],
)
gentbl(
name = "hlo_ops_base_enums_inc_gen",
tbl_outs = [
("-gen-enum-decls", "include/mlir-hlo/Dialect/mhlo/IR/hlo_ops_base_enums.h.inc"),
("-gen-enum-defs", "include/mlir-hlo/Dialect/mhlo/IR/hlo_ops_base_enums.cc.inc"),
],
tblgen = "@llvm-project//mlir:mlir-tblgen",
td_file = "include/mlir-hlo/Dialect/mhlo/IR/hlo_ops_base.td",
td_includes = [
Change hardcoded external/ include to match repository name This is the preferred name for the Bazel repository when included from another project. Note: These hardcoded "external/" includes are a horrible hack that we can hopefully get rid of soon with more robust Bazel tablegen rules. PiperOrigin-RevId: 350810768
2021-01-09 04:06:39 +08:00
"external/mlir-hlo/include",
Add Bazel build configuration to MLIR HLO standalone repo. Tested that this works in OSS in tree with `bazel build //...` and as a submodule in IREE. PiperOrigin-RevId: 347919106
2020-12-17 08:22:14 +08:00
"include",
],
td_relative_includes = [
"include",
],
td_srcs = [":hlo_ops_td_files"],
)
gentbl(
name = "hlo_ops_pattern_gen",
strip_include_prefix = "lib/Dialect/mhlo/IR/",
tbl_outs = [
(
"-gen-rewriters",
"lib/Dialect/mhlo/IR/hlo_patterns.cc.inc",
),
],
tblgen = "@llvm-project//mlir:mlir-tblgen",
td_file = "lib/Dialect/mhlo/IR/hlo_patterns.td",
td_includes = [
Change hardcoded external/ include to match repository name This is the preferred name for the Bazel repository when included from another project. Note: These hardcoded "external/" includes are a horrible hack that we can hopefully get rid of soon with more robust Bazel tablegen rules. PiperOrigin-RevId: 350810768
2021-01-09 04:06:39 +08:00
"external/mlir-hlo/include",
Add Bazel build configuration to MLIR HLO standalone repo. Tested that this works in OSS in tree with `bazel build //...` and as a submodule in IREE. PiperOrigin-RevId: 347919106
2020-12-17 08:22:14 +08:00
"include",
],
td_relative_includes = [
"include",
],
td_srcs = [
":hlo_ops_td_files",
"@llvm-project//mlir:StdOpsTdFiles",
"@llvm-project//mlir:include/mlir/Dialect/Shape/IR/ShapeBase.td",
"@llvm-project//mlir:include/mlir/Dialect/Shape/IR/ShapeOps.td",
],
)
[MLIR:LHLO] Add optional call target arg mapping to LMHLO CustomCall operations. - XLA:HLO -> LMHLO conversion drops all token arguments and return values, however custom calls that users write still expect to get buffer pointers for these token types. - To be able to support this, add an optional call target argument mapping attribute to LMHLO custom calls. When this attribute is present, it indicates the number of arguments and returns that the custom call expects and also indicates which LMHLO arg() or output() maps to which arg or result number of the custom call. PiperOrigin-RevId: 358826664
2021-02-23 00:41:59 +08:00
gentbl(
name = "lhlo_ops_structs_inc_gen",
strip_include_prefix = "include",
tbl_outs = [
("-gen-struct-attr-decls", "include/mlir-hlo/Dialect/mhlo/IR/lhlo_ops_structs.h.inc"),
("-gen-struct-attr-defs", "include/mlir-hlo/Dialect/mhlo/IR/lhlo_ops_structs.cc.inc"),
],
tblgen = "@llvm-project//mlir:mlir-tblgen",
td_file = "include/mlir-hlo/Dialect/mhlo/IR/lhlo_ops_structs.td",
td_includes = [
"external/mlir-hlo/include",
"include",
],
td_relative_includes = [
"include",
],
td_srcs = [
":hlo_ops_td_files",
"include/mlir-hlo/Dialect/mhlo/IR/lhlo_dialect.td",
],
)
Add Bazel build configuration to MLIR HLO standalone repo. Tested that this works in OSS in tree with `bazel build //...` and as a submodule in IREE. PiperOrigin-RevId: 347919106
2020-12-17 08:22:14 +08:00
gentbl(
name = "lhlo_ops_inc_gen",
strip_include_prefix = "include",
tbl_outs = [
("-gen-op-decls", "include/mlir-hlo/Dialect/mhlo/IR/lhlo_ops.h.inc"),
("-gen-op-defs", "include/mlir-hlo/Dialect/mhlo/IR/lhlo_ops.cc.inc"),
],
tblgen = "@llvm-project//mlir:mlir-tblgen",
td_file = "include/mlir-hlo/Dialect/mhlo/IR/lhlo_ops.td",
td_includes = [
Change hardcoded external/ include to match repository name This is the preferred name for the Bazel repository when included from another project. Note: These hardcoded "external/" includes are a horrible hack that we can hopefully get rid of soon with more robust Bazel tablegen rules. PiperOrigin-RevId: 350810768
2021-01-09 04:06:39 +08:00
"external/mlir-hlo/include",
Add Bazel build configuration to MLIR HLO standalone repo. Tested that this works in OSS in tree with `bazel build //...` and as a submodule in IREE. PiperOrigin-RevId: 347919106
2020-12-17 08:22:14 +08:00
"include",
],
td_relative_includes = [
"include",
],
[MLIR:LHLO] Add optional call target arg mapping to LMHLO CustomCall operations. - XLA:HLO -> LMHLO conversion drops all token arguments and return values, however custom calls that users write still expect to get buffer pointers for these token types. - To be able to support this, add an optional call target argument mapping attribute to LMHLO custom calls. When this attribute is present, it indicates the number of arguments and returns that the custom call expects and also indicates which LMHLO arg() or output() maps to which arg or result number of the custom call. PiperOrigin-RevId: 358826664
2021-02-23 00:41:59 +08:00
td_srcs = [
":hlo_ops_td_files",
"include/mlir-hlo/Dialect/mhlo/IR/lhlo_ops_structs.td",
"include/mlir-hlo/Dialect/mhlo/IR/lhlo_dialect.td",
],
Add Bazel build configuration to MLIR HLO standalone repo. Tested that this works in OSS in tree with `bazel build //...` and as a submodule in IREE. PiperOrigin-RevId: 347919106
2020-12-17 08:22:14 +08:00
)
gentbl(
name = "lhlo_gpu_ops_structs_inc_gen",
strip_include_prefix = "include",
tbl_outs = [
("-gen-struct-attr-decls", "include/mlir-hlo/Dialect/mhlo/IR/lhlo_gpu_ops_structs.h.inc"),
("-gen-struct-attr-defs", "include/mlir-hlo/Dialect/mhlo/IR/lhlo_gpu_ops_structs.cc.inc"),
],
tblgen = "@llvm-project//mlir:mlir-tblgen",
td_file = "include/mlir-hlo/Dialect/mhlo/IR/lhlo_gpu_ops_structs.td",
td_includes = [
Change hardcoded external/ include to match repository name This is the preferred name for the Bazel repository when included from another project. Note: These hardcoded "external/" includes are a horrible hack that we can hopefully get rid of soon with more robust Bazel tablegen rules. PiperOrigin-RevId: 350810768
2021-01-09 04:06:39 +08:00
"external/mlir-hlo/include",
Add Bazel build configuration to MLIR HLO standalone repo. Tested that this works in OSS in tree with `bazel build //...` and as a submodule in IREE. PiperOrigin-RevId: 347919106
2020-12-17 08:22:14 +08:00
"include",
],
td_relative_includes = [
"include",
],
td_srcs = [
":hlo_ops_td_files",
"include/mlir-hlo/Dialect/mhlo/IR/lhlo_gpu_ops_base.td",
],
)
gentbl(
name = "lhlo_gpu_ops_enums_inc_gen",
strip_include_prefix = "include",
tbl_outs = [
("-gen-enum-decls", "include/mlir-hlo/Dialect/mhlo/IR/lhlo_gpu_ops_enums.h.inc"),
("-gen-enum-defs", "include/mlir-hlo/Dialect/mhlo/IR/lhlo_gpu_ops_enums.cc.inc"),
],
tblgen = "@llvm-project//mlir:mlir-tblgen",
td_file = "include/mlir-hlo/Dialect/mhlo/IR/lhlo_gpu_ops_enums.td",
td_includes = [
Change hardcoded external/ include to match repository name This is the preferred name for the Bazel repository when included from another project. Note: These hardcoded "external/" includes are a horrible hack that we can hopefully get rid of soon with more robust Bazel tablegen rules. PiperOrigin-RevId: 350810768
2021-01-09 04:06:39 +08:00
"external/mlir-hlo/include",
Add Bazel build configuration to MLIR HLO standalone repo. Tested that this works in OSS in tree with `bazel build //...` and as a submodule in IREE. PiperOrigin-RevId: 347919106
2020-12-17 08:22:14 +08:00
"include",
],
td_relative_includes = [
"include",
],
td_srcs = [":hlo_ops_td_files"],
)
cc_library(
name = "lhlo_gpu_ops_structs",
srcs = [
"include/mlir-hlo/Dialect/mhlo/IR/lhlo_gpu_ops_structs.cc.inc",
"include/mlir-hlo/Dialect/mhlo/IR/lhlo_gpu_ops_structs.h.inc",
"lib/Dialect/mhlo/IR/lhlo_gpu_ops_structs.cc",
],
hdrs = [
"include/mlir-hlo/Dialect/mhlo/IR/lhlo_gpu_ops_structs.h",
],
includes = ["include"],
deps = [
":lhlo_gpu_ops_structs_inc_gen",
"@llvm-project//mlir:IR",
"@llvm-project//mlir:Support",
],
)
cc_library(
name = "lhlo_gpu_ops_enums",
srcs = [
"include/mlir-hlo/Dialect/mhlo/IR/lhlo_gpu_ops_enums.cc.inc",
"include/mlir-hlo/Dialect/mhlo/IR/lhlo_gpu_ops_enums.h.inc",
"lib/Dialect/mhlo/IR/lhlo_gpu_ops_enums.cc",
],
hdrs = [
"include/mlir-hlo/Dialect/mhlo/IR/lhlo_gpu_ops_enums.h",
],
includes = ["include"],
deps = [
":lhlo_gpu_ops_enums_inc_gen",
"@llvm-project//llvm:Support",
],
)
gentbl(
name = "lhlo_gpu_ops_inc_gen",
strip_include_prefix = "include",
tbl_outs = [
("-gen-op-decls", "include/mlir-hlo/Dialect/mhlo/IR/lhlo_gpu_ops.h.inc"),
("-gen-op-defs", "include/mlir-hlo/Dialect/mhlo/IR/lhlo_gpu_ops.cc.inc"),
],
tblgen = "@llvm-project//mlir:mlir-tblgen",
td_file = "include/mlir-hlo/Dialect/mhlo/IR/lhlo_gpu_ops.td",
td_includes = [
Change hardcoded external/ include to match repository name This is the preferred name for the Bazel repository when included from another project. Note: These hardcoded "external/" includes are a horrible hack that we can hopefully get rid of soon with more robust Bazel tablegen rules. PiperOrigin-RevId: 350810768
2021-01-09 04:06:39 +08:00
"external/mlir-hlo/include",
Add Bazel build configuration to MLIR HLO standalone repo. Tested that this works in OSS in tree with `bazel build //...` and as a submodule in IREE. PiperOrigin-RevId: 347919106
2020-12-17 08:22:14 +08:00
"include",
],
td_relative_includes = [
"include",
],
td_srcs = [
":hlo_ops_td_files",
"include/mlir-hlo/Dialect/mhlo/IR/lhlo_gpu_ops_base.td",
"include/mlir-hlo/Dialect/mhlo/IR/lhlo_gpu_ops_structs.td",
"include/mlir-hlo/Dialect/mhlo/IR/lhlo_gpu_ops_enums.td",
],
)
#TODO(aminim): revisit the naming and grouping of these rules post-move.
gentbl(
name = "canonicalize_inc_gen",
strip_include_prefix = "lib/Dialect/mhlo/IR/",
tbl_outs = [
("-gen-rewriters", "lib/Dialect/mhlo/IR/mhlo_canonicalize.inc"),
],
tblgen = "@llvm-project//mlir:mlir-tblgen",
td_file = "lib/Dialect/mhlo/IR/mhlo_canonicalize.td",
td_includes = [
Change hardcoded external/ include to match repository name This is the preferred name for the Bazel repository when included from another project. Note: These hardcoded "external/" includes are a horrible hack that we can hopefully get rid of soon with more robust Bazel tablegen rules. PiperOrigin-RevId: 350810768
2021-01-09 04:06:39 +08:00
"external/mlir-hlo/include",
Add Bazel build configuration to MLIR HLO standalone repo. Tested that this works in OSS in tree with `bazel build //...` and as a submodule in IREE. PiperOrigin-RevId: 347919106
2020-12-17 08:22:14 +08:00
"include",
],
td_relative_includes = [
"include",
],
td_srcs = [":hlo_ops_td_files"],
)
gentbl(
name = "infer_fusibility_op_interface_gen",
tbl_outs = [
(
"-gen-op-interface-decls",
"include/mlir-hlo/Dialect/mhlo/IR/infer_fusibility_op_interface.h.inc",
),
(
"-gen-op-interface-defs",
"include/mlir-hlo/Dialect/mhlo/IR/infer_fusibility_op_interface.cpp.inc",
),
],
tblgen = "@llvm-project//mlir:mlir-tblgen",
td_file = "include/mlir-hlo/Dialect/mhlo/IR/infer_fusibility_op_interface.td",
td_includes = [
Change hardcoded external/ include to match repository name This is the preferred name for the Bazel repository when included from another project. Note: These hardcoded "external/" includes are a horrible hack that we can hopefully get rid of soon with more robust Bazel tablegen rules. PiperOrigin-RevId: 350810768
2021-01-09 04:06:39 +08:00
"external/mlir-hlo/include",
Add Bazel build configuration to MLIR HLO standalone repo. Tested that this works in OSS in tree with `bazel build //...` and as a submodule in IREE. PiperOrigin-RevId: 347919106
2020-12-17 08:22:14 +08:00
"include",
],
td_relative_includes = [
"include",
],
td_srcs = [
":hlo_ops_td_files",
],
)
cc_library(
name = "infer_fusibility_op_interface",
srcs = [
"lib/Dialect/mhlo/IR/infer_fusibility_op_interface.cc",
],
hdrs = [
"include/mlir-hlo/Dialect/mhlo/IR/infer_fusibility_op_interface.h",
"include/mlir-hlo/Dialect/mhlo/IR/infer_fusibility_op_interface.h.inc",
],
includes = ["include"],
deps = [
":infer_fusibility_op_interface_gen",
"@llvm-project//mlir:IR",
"@llvm-project//mlir:Support",
],
alwayslink = 1,
)
cc_library(
name = "hlo_ops_base_structs",
srcs = [
"include/mlir-hlo/Dialect/mhlo/IR/hlo_ops_base_structs.h.inc",
"lib/Dialect/mhlo/IR/hlo_ops_base_structs.cc",
],
hdrs = [
"include/mlir-hlo/Dialect/mhlo/IR/hlo_ops_base_structs.h",
],
includes = ["include"],
deps = [
":hlo_ops_base_structs_inc_gen",
"@llvm-project//mlir:IR",
"@llvm-project//mlir:Support",
],
)
cc_library(
name = "hlo_ops_base_enums",
srcs = [
"include/mlir-hlo/Dialect/mhlo/IR/hlo_ops_base_enums.h.inc",
"lib/Dialect/mhlo/IR/hlo_ops_base_enums.cc",
],
hdrs = [
"include/mlir-hlo/Dialect/mhlo/IR/hlo_ops_base_enums.h",
],
includes = ["include"],
deps = [
":hlo_ops_base_enums_inc_gen",
"@llvm-project//llvm:Support",
],
)
cc_library(
name = "convert_op_folder",
srcs = ["lib/utils/convert_op_folder.cc"],
hdrs = ["include/mlir-hlo/utils/convert_op_folder.h"],
includes = ["include"],
deps = [
"@llvm-project//mlir:IR",
],
)
cc_library(
name = "hlo",
srcs = [
"include/mlir-hlo/Dialect/mhlo/IR/hlo_ops.cc.inc",
"include/mlir-hlo/Dialect/mhlo/IR/hlo_ops.h.inc",
"lib/Dialect/mhlo/IR/chlo_ops.cc",
"lib/Dialect/mhlo/IR/hlo_ops.cc",
"lib/utils/broadcast_utils.cc",
"lib/utils/hlo_utils.cc",
],
hdrs = [
"include/mlir-hlo/Dialect/mhlo/IR/chlo_ops.h",
"include/mlir-hlo/Dialect/mhlo/IR/hlo_ops.h",
"include/mlir-hlo/utils/broadcast_utils.h",
"include/mlir-hlo/utils/hlo_utils.h",
],
includes = ["include"],
deps = [
"hlo_ops_pattern_gen",
":canonicalize_inc_gen",
":chlo_ops_inc_gen",
":convert_op_folder",
":hlo_ops_base_enums",
":hlo_ops_base_inc_gen",
":hlo_ops_base_structs",
":hlo_ops_inc_gen",
":infer_fusibility_op_interface",
"@llvm-project//llvm:Support",
"@llvm-project//mlir:Analysis",
"@llvm-project//mlir:IR",
"@llvm-project//mlir:InferTypeOpInterface",
"@llvm-project//mlir:Pass",
"@llvm-project//mlir:Shape",
"@llvm-project//mlir:SideEffects",
"@llvm-project//mlir:StandardOps",
"@llvm-project//mlir:Support",
Integrate LLVM at llvm/llvm-project@c3acda0798f9 Updates LLVM usage to match [c3acda0798f9](https://github.com/llvm/llvm-project/commit/c3acda0798f9) PiperOrigin-RevId: 348896724
2020-12-24 15:53:08 +08:00
"@llvm-project//mlir:TensorDialect",
Add Bazel build configuration to MLIR HLO standalone repo. Tested that this works in OSS in tree with `bazel build //...` and as a submodule in IREE. PiperOrigin-RevId: 347919106
2020-12-17 08:22:14 +08:00
"@llvm-project//mlir:TransformUtils",
"@llvm-project//mlir:Transforms",
],
alwayslink = 1,
)
cc_library(
name = "lhlo",
srcs = [
"include/mlir-hlo/Dialect/mhlo/IR/lhlo_ops.cc.inc",
"include/mlir-hlo/Dialect/mhlo/IR/lhlo_ops.h.inc",
[MLIR:LHLO] Add optional call target arg mapping to LMHLO CustomCall operations. - XLA:HLO -> LMHLO conversion drops all token arguments and return values, however custom calls that users write still expect to get buffer pointers for these token types. - To be able to support this, add an optional call target argument mapping attribute to LMHLO custom calls. When this attribute is present, it indicates the number of arguments and returns that the custom call expects and also indicates which LMHLO arg() or output() maps to which arg or result number of the custom call. PiperOrigin-RevId: 358826664
2021-02-23 00:41:59 +08:00
"include/mlir-hlo/Dialect/mhlo/IR/lhlo_ops_structs.cc.inc",
"include/mlir-hlo/Dialect/mhlo/IR/lhlo_ops_structs.h",
"include/mlir-hlo/Dialect/mhlo/IR/lhlo_ops_structs.h.inc",
Add Bazel build configuration to MLIR HLO standalone repo. Tested that this works in OSS in tree with `bazel build //...` and as a submodule in IREE. PiperOrigin-RevId: 347919106
2020-12-17 08:22:14 +08:00
"lib/Dialect/mhlo/IR/lhlo_ops.cc",
[MLIR:LHLO] Add optional call target arg mapping to LMHLO CustomCall operations. - XLA:HLO -> LMHLO conversion drops all token arguments and return values, however custom calls that users write still expect to get buffer pointers for these token types. - To be able to support this, add an optional call target argument mapping attribute to LMHLO custom calls. When this attribute is present, it indicates the number of arguments and returns that the custom call expects and also indicates which LMHLO arg() or output() maps to which arg or result number of the custom call. PiperOrigin-RevId: 358826664
2021-02-23 00:41:59 +08:00
"lib/Dialect/mhlo/IR/lhlo_ops_structs.cc",
Add Bazel build configuration to MLIR HLO standalone repo. Tested that this works in OSS in tree with `bazel build //...` and as a submodule in IREE. PiperOrigin-RevId: 347919106
2020-12-17 08:22:14 +08:00
],
hdrs = [
"include/mlir-hlo/Dialect/mhlo/IR/lhlo_ops.h",
],
includes = ["include"],
deps = [
":hlo_ops_base_enums",
":hlo_ops_base_inc_gen",
":hlo_ops_base_structs",
":lhlo_ops_inc_gen",
"@llvm-project//llvm:Support",
"@llvm-project//mlir:Analysis",
"@llvm-project//mlir:CopyOpInterface",
"@llvm-project//mlir:IR",
"@llvm-project//mlir:Pass",
"@llvm-project//mlir:SideEffects",
"@llvm-project//mlir:StandardOps",
"@llvm-project//mlir:Support",
"@llvm-project//mlir:TransformUtils",
"@llvm-project//mlir:Transforms",
"@llvm-project//mlir:ViewLikeInterface",
],
alwayslink = 1,
)
cc_library(
name = "lhlo_gpu",
srcs = [
"include/mlir-hlo/Dialect/mhlo/IR/lhlo_gpu_ops.cc.inc",
"include/mlir-hlo/Dialect/mhlo/IR/lhlo_gpu_ops.h.inc",
"lib/Dialect/mhlo/IR/lhlo_gpu_ops.cc",
],
hdrs = [
"include/mlir-hlo/Dialect/mhlo/IR/lhlo_gpu_ops.h",
],
includes = ["include"],
deps = [
":hlo",
":hlo_ops_base_enums",
":hlo_ops_base_structs",
":infer_fusibility_op_interface",
":lhlo_gpu_ops_enums",
":lhlo_gpu_ops_inc_gen",
":lhlo_gpu_ops_structs",
"@llvm-project//llvm:Support",
"@llvm-project//mlir:Analysis",
"@llvm-project//mlir:CopyOpInterface",
"@llvm-project//mlir:IR",
"@llvm-project//mlir:InferTypeOpInterface",
"@llvm-project//mlir:Pass",
"@llvm-project//mlir:SideEffects",
"@llvm-project//mlir:StandardOps",
"@llvm-project//mlir:Support",
"@llvm-project//mlir:TransformUtils",
"@llvm-project//mlir:Transforms",
"@llvm-project//mlir:ViewLikeInterface",
],
alwayslink = 1,
)
cc_library(
name = "hlo_dialect_registration",
srcs = ["lib/Dialect/mhlo/IR/init.cc"],
hdrs = ["include/mlir-hlo/Dialect/mhlo/IR/register.h"],
deps = [
":hlo",
":lhlo",
":lhlo_gpu",
"@llvm-project//mlir:IR",
],
)
cc_library(
name = "sink_constants_to_control_flow",
srcs = [
"lib/Dialect/mhlo/transforms/sink_constants_to_control_flow.cc",
],
hdrs = ["include/mlir-hlo/Dialect/mhlo/transforms/passes.h"],
deps = [
":hlo",
":pass_details",
"@llvm-project//llvm:Support",
"@llvm-project//mlir:IR",
"@llvm-project//mlir:Pass",
"@llvm-project//mlir:SCFDialect",
"@llvm-project//mlir:StandardOps",
"@llvm-project//mlir:Support",
"@llvm-project//mlir:Transforms",
],
alwayslink = 1,
)
cc_library(
name = "mhlo_control_flow_to_scf",
srcs = ["lib/Dialect/mhlo/transforms/mhlo_control_flow_to_scf.cc"],
hdrs = ["include/mlir-hlo/Dialect/mhlo/transforms/passes.h"],
deps = [
":hlo",
"@llvm-project//llvm:Support",
"@llvm-project//mlir:IR",
"@llvm-project//mlir:Pass",
"@llvm-project//mlir:SCFDialect",
"@llvm-project//mlir:StandardOps",
"@llvm-project//mlir:Support",
Integrate LLVM at llvm/llvm-project@0cf7e4b252fe Updates LLVM usage to match [0cf7e4b252fe](https://github.com/llvm/llvm-project/commit/0cf7e4b252fe) PiperOrigin-RevId: 347948887
2020-12-17 12:29:15 +08:00
"@llvm-project//mlir:TensorDialect",
Add Bazel build configuration to MLIR HLO standalone repo. Tested that this works in OSS in tree with `bazel build //...` and as a submodule in IREE. PiperOrigin-RevId: 347919106
2020-12-17 08:22:14 +08:00
],
)
cc_library(
name = "map_lmhlo_to_scalar_op",
hdrs = ["include/mlir-hlo/Dialect/mhlo/transforms/map_lmhlo_to_scalar_op.h"],
deps = [
":hlo",
":lhlo",
":map_hlo_to_lhlo_op",
"@llvm-project//llvm:Support",
[MLIR] Migrate TF from STD complex ops to ComplexDialect. PiperOrigin-RevId: 352966408
2021-01-21 17:21:23 +08:00
"@llvm-project//mlir:ComplexDialect",
Add Bazel build configuration to MLIR HLO standalone repo. Tested that this works in OSS in tree with `bazel build //...` and as a submodule in IREE. PiperOrigin-RevId: 347919106
2020-12-17 08:22:14 +08:00
"@llvm-project//mlir:IR",
Integrate LLVM at llvm/llvm-project@2bfe27da171e Updates LLVM usage to match [2bfe27da171e](https://github.com/llvm/llvm-project/commit/2bfe27da171e) PiperOrigin-RevId: 357196336
2021-02-13 00:30:51 +08:00
"@llvm-project//mlir:MathDialect",
Define lowering of [l]mhlo.pow. For floating point operations, this uses std.pow. For integer operations, this lowers to a loop. This adds a dependency on scf. PiperOrigin-RevId: 348537232
2020-12-22 07:26:38 +08:00
"@llvm-project//mlir:SCFDialect",
Add Bazel build configuration to MLIR HLO standalone repo. Tested that this works in OSS in tree with `bazel build //...` and as a submodule in IREE. PiperOrigin-RevId: 347919106
2020-12-17 08:22:14 +08:00
"@llvm-project//mlir:StandardOps",
],
)
cc_library(
name = "map_chlo_to_hlo_op",
hdrs = ["include/mlir-hlo/Dialect/mhlo/transforms/map_chlo_to_hlo_op.h"],
deps = [
":hlo",
"@llvm-project//mlir:IR",
],
)
cc_library(
name = "map_hlo_to_lhlo_op",
hdrs = ["include/mlir-hlo/Dialect/mhlo/transforms/map_hlo_to_lhlo_op.h"],
deps = [
":hlo",
":lhlo",
],
)
cc_library(
name = "lhlo_legalize_to_affine",
srcs = ["lib/Dialect/mhlo/transforms/lhlo_legalize_to_affine.cc"],
deps = [
":hlo",
":lhlo",
":map_lmhlo_to_scalar_op",
"@llvm-project//llvm:Support",
"@llvm-project//mlir:Affine",
"@llvm-project//mlir:IR",
"@llvm-project//mlir:Pass",
"@llvm-project//mlir:StandardOps",
"@llvm-project//mlir:TransformUtils",
],
alwayslink = 1,
)
cc_library(
name = "lhlo_legalize_to_parallel_loops",
srcs = ["lib/Dialect/mhlo/transforms/lhlo_legalize_to_parallel_loops.cc"],
deps = [
":lhlo",
"@llvm-project//llvm:Support",
"@llvm-project//mlir:IR",
"@llvm-project//mlir:LinalgOps",
"@llvm-project//mlir:Pass",
"@llvm-project//mlir:SCFDialect",
"@llvm-project//mlir:StandardOps",
"@llvm-project//mlir:Transforms",
],
alwayslink = 1,
)
cc_library(
name = "legalize_to_linalg",
srcs = ["lib/Dialect/mhlo/transforms/legalize_to_linalg.cc"],
hdrs = [
"include/mlir-hlo/Dialect/mhlo/transforms/passes.h",
"include/mlir-hlo/Dialect/mhlo/transforms/rewriters.h",
],
deps = [
":hlo",
":lhlo",
":map_lmhlo_to_scalar_op",
"@llvm-project//llvm:Support",
"@llvm-project//mlir:Affine",
"@llvm-project//mlir:IR",
"@llvm-project//mlir:LinalgOps",
Integrate LLVM at llvm/llvm-project@2bfe27da171e Updates LLVM usage to match [2bfe27da171e](https://github.com/llvm/llvm-project/commit/2bfe27da171e) PiperOrigin-RevId: 357196336
2021-02-13 00:30:51 +08:00
"@llvm-project//mlir:MathDialect",
Add Bazel build configuration to MLIR HLO standalone repo. Tested that this works in OSS in tree with `bazel build //...` and as a submodule in IREE. PiperOrigin-RevId: 347919106
2020-12-17 08:22:14 +08:00
"@llvm-project//mlir:Pass",
Define lowering of [l]mhlo.pow. For floating point operations, this uses std.pow. For integer operations, this lowers to a loop. This adds a dependency on scf. PiperOrigin-RevId: 348537232
2020-12-22 07:26:38 +08:00
"@llvm-project//mlir:SCFDialect",
Add Bazel build configuration to MLIR HLO standalone repo. Tested that this works in OSS in tree with `bazel build //...` and as a submodule in IREE. PiperOrigin-RevId: 347919106
2020-12-17 08:22:14 +08:00
"@llvm-project//mlir:StandardOps",
Lower integer matmuls to linalg PiperOrigin-RevId: 359306495
2021-02-25 01:43:54 +08:00
"@llvm-project//mlir:Support",
[KERNEL_GEN] Add a pattern for hlo.dyn_broadcast->linalg to enable is_inf kernel. PiperOrigin-RevId: 351179620
2021-01-12 02:11:39 +08:00
"@llvm-project//mlir:TensorDialect",
Add Bazel build configuration to MLIR HLO standalone repo. Tested that this works in OSS in tree with `bazel build //...` and as a submodule in IREE. PiperOrigin-RevId: 347919106
2020-12-17 08:22:14 +08:00
"@llvm-project//mlir:Transforms",
],
alwayslink = 1,
)
cc_library(
name = "transform_unranked_hlo",
srcs = ["lib/Dialect/mhlo/transforms/transform_unranked_hlo.cc"],
hdrs = [
"include/mlir-hlo/Dialect/mhlo/transforms/passes.h",
"include/mlir-hlo/Dialect/mhlo/transforms/rewriters.h",
],
deps = [
":hlo",
":map_chlo_to_hlo_op",
"@llvm-project//llvm:Support",
"@llvm-project//mlir:IR",
"@llvm-project//mlir:Pass",
"@llvm-project//mlir:SCFDialect",
"@llvm-project//mlir:Shape",
"@llvm-project//mlir:StandardOps",
Integrate LLVM at llvm/llvm-project@c3acda0798f9 Updates LLVM usage to match [c3acda0798f9](https://github.com/llvm/llvm-project/commit/c3acda0798f9) PiperOrigin-RevId: 348896724
2020-12-24 15:53:08 +08:00
"@llvm-project//mlir:TensorDialect",
Add Bazel build configuration to MLIR HLO standalone repo. Tested that this works in OSS in tree with `bazel build //...` and as a submodule in IREE. PiperOrigin-RevId: 347919106
2020-12-17 08:22:14 +08:00
"@llvm-project//mlir:Transforms",
],
alwayslink = 1,
)
cc_library(
name = "lhlo_legalize_to_gpu",
srcs = ["lib/Dialect/mhlo/transforms/lhlo_legalize_to_gpu.cc"],
deps = [
":hlo",
":lhlo",
":map_lmhlo_to_scalar_op",
"@llvm-project//llvm:Support",
"@llvm-project//mlir:Affine",
"@llvm-project//mlir:GPUDialect",
"@llvm-project//mlir:IR",
"@llvm-project//mlir:LinalgOps",
"@llvm-project//mlir:Pass",
"@llvm-project//mlir:SCFDialect",
"@llvm-project//mlir:StandardOps",
"@llvm-project//mlir:Transforms",
],
alwayslink = 1,
)
cc_library(
name = "lhlo_fuse_linalg",
srcs = ["lib/Dialect/mhlo/transforms/lhlo_fuse_linalg.cc"],
hdrs = ["include/mlir-hlo/Dialect/mhlo/transforms/passes.h"],
deps = [
":lhlo",
"@llvm-project//llvm:Support",
"@llvm-project//mlir:Affine",
"@llvm-project//mlir:IR",
"@llvm-project//mlir:LinalgTransforms",
"@llvm-project//mlir:Pass",
"@llvm-project//mlir:SCFDialect",
"@llvm-project//mlir:StandardOps",
"@llvm-project//mlir:Support",
Integrate LLVM at llvm/llvm-project@c3acda0798f9 Updates LLVM usage to match [c3acda0798f9](https://github.com/llvm/llvm-project/commit/c3acda0798f9) PiperOrigin-RevId: 348896724
2020-12-24 15:53:08 +08:00
"@llvm-project//mlir:TensorDialect",
Add Bazel build configuration to MLIR HLO standalone repo. Tested that this works in OSS in tree with `bazel build //...` and as a submodule in IREE. PiperOrigin-RevId: 347919106
2020-12-17 08:22:14 +08:00
"@llvm-project//mlir:TransformUtils",
"@llvm-project//mlir:ViewLikeInterface",
],
alwayslink = 1,
)
cc_library(
name = "hlo_legalize_to_lhlo",
srcs = ["lib/Dialect/mhlo/transforms/hlo_legalize_to_lhlo.cc"],
hdrs = [
"include/mlir-hlo/Dialect/mhlo/transforms/passes.h",
"include/mlir-hlo/Dialect/mhlo/transforms/rewriters.h",
],
deps = [
":hlo",
":lhlo",
":map_hlo_to_lhlo_op",
"@llvm-project//llvm:Support",
"@llvm-project//mlir:IR",
"@llvm-project//mlir:Pass",
"@llvm-project//mlir:Shape",
"@llvm-project//mlir:ShapeTransforms",
"@llvm-project//mlir:StandardOps",
"@llvm-project//mlir:StandardOpsTransforms",
"@llvm-project//mlir:Support",
Integrate LLVM at llvm/llvm-project@0cf7e4b252fe Updates LLVM usage to match [0cf7e4b252fe](https://github.com/llvm/llvm-project/commit/0cf7e4b252fe) PiperOrigin-RevId: 347948887
2020-12-17 12:29:15 +08:00
"@llvm-project//mlir:TensorDialect",
Add Bazel build configuration to MLIR HLO standalone repo. Tested that this works in OSS in tree with `bazel build //...` and as a submodule in IREE. PiperOrigin-RevId: 347919106
2020-12-17 08:22:14 +08:00
"@llvm-project//mlir:Transforms",
],
alwayslink = 1,
)
cc_library(
name = "cycle_detector",
srcs = ["lib/utils/cycle_detector.cc"],
hdrs = ["include/mlir-hlo/utils/cycle_detector.h"],
includes = ["include"],
deps = [
"@llvm-project//llvm:Support",
],
alwayslink = 1,
)
cc_library(
name = "mhlo_fusion",
srcs = ["lib/Dialect/mhlo/transforms/mhlo_fusion.cc"],
deps = [
":cycle_detector",
":hlo",
"@llvm-project//llvm:Core",
"@llvm-project//llvm:Support",
"@llvm-project//mlir:IR",
"@llvm-project//mlir:Pass",
"@llvm-project//mlir:StandardOps",
"@llvm-project//mlir:Support",
"@llvm-project//mlir:TransformUtils",
],
alwayslink = 1,
)
gentbl(
name = "legalize_to_standard_inc_gen",
strip_include_prefix = "lib/Dialect/mhlo/transforms/",
tbl_outs = [
("-gen-rewriters", "lib/Dialect/mhlo/transforms/generated_legalize_to_standard.inc"),
],
tblgen = "@llvm-project//mlir:mlir-tblgen",
td_file = "lib/Dialect/mhlo/transforms/legalize_to_standard_patterns.td",
td_includes = [
Change hardcoded external/ include to match repository name This is the preferred name for the Bazel repository when included from another project. Note: These hardcoded "external/" includes are a horrible hack that we can hopefully get rid of soon with more robust Bazel tablegen rules. PiperOrigin-RevId: 350810768
2021-01-09 04:06:39 +08:00
"external/mlir-hlo/include",
Add Bazel build configuration to MLIR HLO standalone repo. Tested that this works in OSS in tree with `bazel build //...` and as a submodule in IREE. PiperOrigin-RevId: 347919106
2020-12-17 08:22:14 +08:00
"include",
],
td_relative_includes = [
"include",
],
td_srcs = [
":hlo_ops_td_files",
"@llvm-project//mlir:StdOpsTdFiles",
],
)
cc_library(
name = "legalize_control_flow",
srcs = ["lib/Dialect/mhlo/transforms/legalize_control_flow.cc"],
hdrs = ["include/mlir-hlo/Dialect/mhlo/transforms/passes.h"],
deps = [
":hlo",
"@llvm-project//llvm:Support",
"@llvm-project//mlir:IR",
"@llvm-project//mlir:Pass",
"@llvm-project//mlir:StandardOps",
"@llvm-project//mlir:Support",
Integrate LLVM at llvm/llvm-project@0cf7e4b252fe Updates LLVM usage to match [0cf7e4b252fe](https://github.com/llvm/llvm-project/commit/0cf7e4b252fe) PiperOrigin-RevId: 347948887
2020-12-17 12:29:15 +08:00
"@llvm-project//mlir:TensorDialect",
Add Bazel build configuration to MLIR HLO standalone repo. Tested that this works in OSS in tree with `bazel build //...` and as a submodule in IREE. PiperOrigin-RevId: 347919106
2020-12-17 08:22:14 +08:00
],
alwayslink = 1,
)
cc_library(
name = "legalize_to_standard",
srcs = ["lib/Dialect/mhlo/transforms/legalize_to_standard.cc"],
hdrs = ["include/mlir-hlo/Dialect/mhlo/transforms/passes.h"],
deps = [
":hlo",
":legalize_to_standard_inc_gen",
":legalize_trigonometric_to_approximation",
"@llvm-project//llvm:Support",
"@llvm-project//mlir:IR",
"@llvm-project//mlir:Pass",
"@llvm-project//mlir:StandardOps",
"@llvm-project//mlir:Support",
"@llvm-project//mlir:TransformUtils",
],
alwayslink = 1,
)
cc_library(
name = "legalize_gather_to_torch_index_select",
srcs = ["lib/Dialect/mhlo/transforms/legalize_gather_to_torch_index_select.cc"],
hdrs = [
"include/mlir-hlo/Dialect/mhlo/transforms/passes.h",
"include/mlir-hlo/Dialect/mhlo/transforms/rewriters.h",
],
deps = [
":hlo",
"@llvm-project//llvm:Support",
"@llvm-project//mlir:IR",
"@llvm-project//mlir:Pass",
"@llvm-project//mlir:StandardOps",
"@llvm-project//mlir:Support",
"@llvm-project//mlir:Transforms",
],
alwayslink = 1,
)
cc_library(
name = "legalize_trigonometric_to_approximation",
srcs = ["lib/Dialect/mhlo/transforms/legalize_trigonometric_to_approximation.cc"],
hdrs = [
"include/mlir-hlo/Dialect/mhlo/transforms/passes.h",
"include/mlir-hlo/Dialect/mhlo/transforms/rewriters.h",
],
includes = ["include"],
deps = [
"@llvm-project//llvm:Support",
"@llvm-project//mlir:IR",
Integrate LLVM at llvm/llvm-project@2bfe27da171e Updates LLVM usage to match [2bfe27da171e](https://github.com/llvm/llvm-project/commit/2bfe27da171e) PiperOrigin-RevId: 357196336
2021-02-13 00:30:51 +08:00
"@llvm-project//mlir:MathDialect",
Add Bazel build configuration to MLIR HLO standalone repo. Tested that this works in OSS in tree with `bazel build //...` and as a submodule in IREE. PiperOrigin-RevId: 347919106
2020-12-17 08:22:14 +08:00
"@llvm-project//mlir:Pass",
"@llvm-project//mlir:StandardOps",
"@llvm-project//mlir:Support",
"@llvm-project//mlir:Transforms",
],
alwayslink = 1,
)
gentbl(
name = "lower_complex_inc_gen",
strip_include_prefix = "lib/Dialect/mhlo/transforms/",
tbl_outs = [
("-gen-rewriters", "lib/Dialect/mhlo/transforms/generated_lower_complex.inc"),
],
tblgen = "@llvm-project//mlir:mlir-tblgen",
td_file = "lib/Dialect/mhlo/transforms/lower_complex_patterns.td",
td_includes = [
Change hardcoded external/ include to match repository name This is the preferred name for the Bazel repository when included from another project. Note: These hardcoded "external/" includes are a horrible hack that we can hopefully get rid of soon with more robust Bazel tablegen rules. PiperOrigin-RevId: 350810768
2021-01-09 04:06:39 +08:00
"external/mlir-hlo/include",
Add Bazel build configuration to MLIR HLO standalone repo. Tested that this works in OSS in tree with `bazel build //...` and as a submodule in IREE. PiperOrigin-RevId: 347919106
2020-12-17 08:22:14 +08:00
"include",
],
td_relative_includes = [
"include",
],
td_srcs = [
":hlo_ops_td_files",
"@llvm-project//llvm:Support",
"@llvm-project//mlir:StdOpsTdFiles",
],
)
cc_library(
#TODO(aminim): find a better name here?
name = "mhlo_to_mhlo_lowering_patterns",
srcs = [
"lib/Dialect/mhlo/transforms/lower_complex.cc",
"lib/Dialect/mhlo/transforms/lower_general_dot.cc",
"lib/Dialect/mhlo/transforms/optimize_mhlo.cc",
],
hdrs = [
"include/mlir-hlo/Dialect/mhlo/transforms/passes.h",
"include/mlir-hlo/Dialect/mhlo/transforms/rewriters.h",
],
deps = [
":hlo",
":lower_complex_inc_gen",
"@llvm-project//llvm:Support",
"@llvm-project//mlir:Analysis",
"@llvm-project//mlir:IR",
"@llvm-project//mlir:Pass",
"@llvm-project//mlir:StandardOps",
"@llvm-project//mlir:Support",
"@llvm-project//mlir:Transforms",
],
alwayslink = 1,
)
cc_library(
name = "materialize_broadcasts",
srcs = [
"lib/Dialect/mhlo/transforms/materialize_broadcasts.cc",
],
hdrs = [
"include/mlir-hlo/Dialect/mhlo/transforms/passes.h",
"include/mlir-hlo/Dialect/mhlo/transforms/rewriters.h",
],
deps = [
":hlo",
"@llvm-project//llvm:Support",
"@llvm-project//mlir:IR",
"@llvm-project//mlir:StandardOps",
"@llvm-project//mlir:Transforms",
],
)
cc_library(
name = "unfuse_batch_norm",
srcs = ["lib/Dialect/mhlo/transforms/unfuse_batch_norm.cc"],
hdrs = [
"include/mlir-hlo/Dialect/mhlo/transforms/passes.h",
],
deps = [
":hlo",
"@llvm-project//llvm:Support",
"@llvm-project//mlir:IR",
"@llvm-project//mlir:StandardOps",
Integrate LLVM at llvm/llvm-project@96ef4f307df2 Updates LLVM usage to match [96ef4f307df2](https://github.com/llvm/llvm-project/commit/96ef4f307df2) PiperOrigin-RevId: 352786460
2021-01-20 23:08:32 +08:00
"@llvm-project//mlir:TensorDialect",
Add Bazel build configuration to MLIR HLO standalone repo. Tested that this works in OSS in tree with `bazel build //...` and as a submodule in IREE. PiperOrigin-RevId: 347919106
2020-12-17 08:22:14 +08:00
"@llvm-project//mlir:Transforms",
],
)
cc_library(
name = "chlo_legalize_to_hlo",
srcs = ["lib/Dialect/mhlo/transforms/chlo_legalize_to_hlo.cc"],
hdrs = ["include/mlir-hlo/Dialect/mhlo/transforms/rewriters.h"],
deps = [
":chlo_legalize_to_hlo_inc_gen",
":hlo",
":map_chlo_to_hlo_op",
[MLIR][CHLO] Generalize lowering with upcast to n-ary operation Allows reuse for zeta lowering now and for the polygamma lowering soon. PiperOrigin-RevId: 357739910
2021-02-17 01:46:33 +08:00
"@llvm-project//llvm:Support",
Add Bazel build configuration to MLIR HLO standalone repo. Tested that this works in OSS in tree with `bazel build //...` and as a submodule in IREE. PiperOrigin-RevId: 347919106
2020-12-17 08:22:14 +08:00
"@llvm-project//mlir:IR",
"@llvm-project//mlir:SCFDialect",
"@llvm-project//mlir:Shape",
"@llvm-project//mlir:StandardOps",
Integrate LLVM at llvm/llvm-project@c3acda0798f9 Updates LLVM usage to match [c3acda0798f9](https://github.com/llvm/llvm-project/commit/c3acda0798f9) PiperOrigin-RevId: 348896724
2020-12-24 15:53:08 +08:00
"@llvm-project//mlir:TensorDialect",
Add Bazel build configuration to MLIR HLO standalone repo. Tested that this works in OSS in tree with `bazel build //...` and as a submodule in IREE. PiperOrigin-RevId: 347919106
2020-12-17 08:22:14 +08:00
"@llvm-project//mlir:Transforms",
],
)
gentbl(
name = "chlo_legalize_to_hlo_inc_gen",
strip_include_prefix = "lib/Dialect/mhlo/transforms/",
tbl_outs = [
(
"-gen-rewriters",
"lib/Dialect/mhlo/transforms/generated_chlo_legalize_to_hlo.inc",
),
],
tblgen = "@llvm-project//mlir:mlir-tblgen",
td_file = "lib/Dialect/mhlo/transforms/chlo_legalize_to_hlo_patterns.td",
td_includes = [
Change hardcoded external/ include to match repository name This is the preferred name for the Bazel repository when included from another project. Note: These hardcoded "external/" includes are a horrible hack that we can hopefully get rid of soon with more robust Bazel tablegen rules. PiperOrigin-RevId: 350810768
2021-01-09 04:06:39 +08:00
"external/mlir-hlo/include",
Add Bazel build configuration to MLIR HLO standalone repo. Tested that this works in OSS in tree with `bazel build //...` and as a submodule in IREE. PiperOrigin-RevId: 347919106
2020-12-17 08:22:14 +08:00
"include",
],
td_relative_includes = [
"include",
],
td_srcs = [
":hlo_ops_td_files",
],
)
cc_library(
name = "pass_details",
hdrs = [
"include/mlir-hlo/Dialect/mhlo/transforms/PassDetail.h",
],
visibility = [
"//visibility:private", # This target is a private detail of pass implementations
],
deps = [
":MhloPassIncGen",
"@llvm-project//mlir:Pass",
],
)
cc_library(
name = "test_passes",
srcs = [
"include/mlir-hlo/Dialect/mhlo/transforms/rewriters.h",
"lib/Dialect/mhlo/transforms/chlo_legalize_to_hlo_pass.cc",
"lib/Dialect/mhlo/transforms/materialize_broadcasts_pass.cc",
"lib/Dialect/mhlo/transforms/optimize_mhlo_pass.cc",
"lib/Dialect/mhlo/transforms/test_infer_shaped_type_pass.cc",
"lib/Dialect/mhlo/transforms/unfuse_batch_norm_pass.cc",
],
deps = [
":chlo_legalize_to_hlo", # build-cleaner: keep
":hlo",
":lhlo",
":materialize_broadcasts", # build-cleaner: keep
":pass_details",
":unfuse_batch_norm", # build-cleaner: keep
"@llvm-project//mlir:IR",
"@llvm-project//mlir:InferTypeOpInterface",
"@llvm-project//mlir:LLVMDialect",
"@llvm-project//mlir:LLVMTransforms",
"@llvm-project//mlir:Pass",
"@llvm-project//mlir:SCFDialect",
"@llvm-project//mlir:Shape",
"@llvm-project//mlir:StandardOps",
Integrate LLVM at llvm/llvm-project@c3acda0798f9 Updates LLVM usage to match [c3acda0798f9](https://github.com/llvm/llvm-project/commit/c3acda0798f9) PiperOrigin-RevId: 348896724
2020-12-24 15:53:08 +08:00
"@llvm-project//mlir:TensorDialect",
Add Bazel build configuration to MLIR HLO standalone repo. Tested that this works in OSS in tree with `bazel build //...` and as a submodule in IREE. PiperOrigin-RevId: 347919106
2020-12-17 08:22:14 +08:00
"@llvm-project//mlir:Transforms",
],
alwayslink = 1,
)
cc_library(
name = "all_passes",
hdrs = [
"include/mlir-hlo/Dialect/mhlo/transforms/register_passes.h",
],
deps = [
":LmhloPassIncGen",
":MhloPassIncGen",
":chlo_legalize_to_hlo",
":hlo_legalize_to_lhlo",
":legalize_control_flow",
":legalize_gather_to_torch_index_select",
":legalize_to_linalg",
":legalize_to_standard",
":legalize_trigonometric_to_approximation",
":lhlo",
":lhlo_fuse_linalg",
":lhlo_legalize_to_affine",
":lhlo_legalize_to_gpu",
":lhlo_legalize_to_parallel_loops",
":mhlo_control_flow_to_scf",
":mhlo_fusion",
":mhlo_to_mhlo_lowering_patterns",
":sink_constants_to_control_flow",
":test_passes",
":transform_unranked_hlo",
"@llvm-project//mlir:Pass",
],
)
cc_binary(
name = "mlir-hlo-opt",
srcs = [
"tools/mlir-hlo-opt/mlir-hlo-opt.cpp",
],
deps = [
":all_passes",
":hlo",
":lhlo",
":lhlo_gpu",
"@llvm-project//llvm:Support",
"@llvm-project//mlir:AllPassesAndDialectsNoRegistration",
"@llvm-project//mlir:IR",
"@llvm-project//mlir:MlirOptLib",
"@llvm-project//mlir:Pass",
"@llvm-project//mlir:Support",
],
)