2021-05-13 01:58:02 +08:00
|
|
|
load("@llvm-project//mlir:tblgen.bzl", "gentbl_cc_library", "gentbl_filegroup", "td_library")
|
2020-12-17 08:22:14 +08:00
|
|
|
|
|
|
|
package(
|
|
|
|
default_visibility = ["//visibility:public"],
|
2021-03-31 08:54:53 +08:00
|
|
|
licenses = ["notice"],
|
2020-12-17 08:22:14 +08:00
|
|
|
)
|
|
|
|
|
2021-03-04 08:10:20 +08:00
|
|
|
exports_files([
|
|
|
|
"include/mlir-hlo/Dialect/mhlo/IR/hlo_ops.td",
|
|
|
|
"include/mlir-hlo/Dialect/mhlo/IR/lhlo_ops.td",
|
|
|
|
])
|
2020-12-17 08:22:14 +08:00
|
|
|
|
2021-03-04 08:10:20 +08:00
|
|
|
td_library(
|
2020-12-17 08:22:14 +08:00
|
|
|
name = "hlo_ops_td_files",
|
2021-03-04 08:10:20 +08:00
|
|
|
srcs = glob(["include/mlir-hlo/Dialect/mhlo/IR/*.td"]) + [
|
|
|
|
# TODO(gcmn): These should be encapsulate in a td_library.
|
2021-05-05 15:26:46 +08:00
|
|
|
"@llvm-project//mlir:include/mlir/Interfaces/ControlFlowInterfaces.td",
|
2020-12-17 08:22:14 +08:00
|
|
|
"@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",
|
2021-03-04 08:10:20 +08:00
|
|
|
"@llvm-project//mlir:include/mlir/Dialect/Shape/IR/ShapeBase.td",
|
|
|
|
"@llvm-project//mlir:include/mlir/Dialect/Shape/IR/ShapeOps.td",
|
2020-12-17 08:22:14 +08:00
|
|
|
],
|
2021-03-04 08:10:20 +08:00
|
|
|
includes = ["include"],
|
|
|
|
deps = [
|
2021-05-05 15:26:46 +08:00
|
|
|
"@llvm-project//mlir:ControlFlowInterfacesTdFiles",
|
|
|
|
"@llvm-project//mlir:LoopLikeInterfaceTdFiles",
|
2021-03-17 04:31:59 +08:00
|
|
|
"@llvm-project//mlir:MemRefOpsTdFiles",
|
2021-03-04 08:10:20 +08:00
|
|
|
"@llvm-project//mlir:OpBaseTdFiles",
|
|
|
|
"@llvm-project//mlir:SideEffectTdFiles",
|
2020-12-17 08:22:14 +08:00
|
|
|
],
|
|
|
|
)
|
|
|
|
|
2021-05-13 01:58:02 +08:00
|
|
|
gentbl_cc_library(
|
2020-12-17 08:22:14 +08:00
|
|
|
name = "MhloPassIncGen",
|
|
|
|
strip_include_prefix = "include",
|
|
|
|
tbl_outs = [
|
|
|
|
(
|
2021-05-13 01:58:02 +08:00
|
|
|
[
|
|
|
|
"-gen-pass-decls",
|
|
|
|
"-name=MHLO",
|
|
|
|
],
|
2020-12-17 08:22:14 +08:00
|
|
|
"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 = [
|
2021-01-09 04:06:39 +08:00
|
|
|
"external/mlir-hlo/include",
|
2020-12-17 08:22:14 +08:00
|
|
|
"include",
|
|
|
|
],
|
2021-03-04 08:10:20 +08:00
|
|
|
deps = [
|
2020-12-17 08:22:14 +08:00
|
|
|
"@llvm-project//mlir:PassBaseTdFiles",
|
|
|
|
],
|
|
|
|
)
|
|
|
|
|
2021-05-13 01:58:02 +08:00
|
|
|
gentbl_cc_library(
|
2020-12-17 08:22:14 +08:00
|
|
|
name = "LmhloPassIncGen",
|
|
|
|
strip_include_prefix = "include",
|
|
|
|
tbl_outs = [
|
|
|
|
(
|
2021-05-13 01:58:02 +08:00
|
|
|
[
|
|
|
|
"-gen-pass-decls",
|
|
|
|
"-name=LMHLO",
|
|
|
|
],
|
2020-12-17 08:22:14 +08:00
|
|
|
"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 = [
|
2021-01-09 04:06:39 +08:00
|
|
|
"external/mlir-hlo/include",
|
2020-12-17 08:22:14 +08:00
|
|
|
"include",
|
|
|
|
],
|
2021-03-04 08:10:20 +08:00
|
|
|
deps = [
|
2020-12-17 08:22:14 +08:00
|
|
|
"@llvm-project//mlir:PassBaseTdFiles",
|
|
|
|
],
|
|
|
|
)
|
|
|
|
|
2021-05-13 01:58:02 +08:00
|
|
|
gentbl_cc_library(
|
2020-12-17 08:22:14 +08:00
|
|
|
name = "chlo_ops_inc_gen",
|
|
|
|
strip_include_prefix = "include",
|
|
|
|
tbl_outs = [
|
2021-05-13 01:58:02 +08:00
|
|
|
(
|
|
|
|
["-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",
|
|
|
|
),
|
2020-12-17 08:22:14 +08:00
|
|
|
],
|
|
|
|
tblgen = "@llvm-project//mlir:mlir-tblgen",
|
|
|
|
td_file = "include/mlir-hlo/Dialect/mhlo/IR/chlo_ops.td",
|
|
|
|
td_includes = [
|
2021-01-09 04:06:39 +08:00
|
|
|
"external/mlir-hlo/include",
|
2020-12-17 08:22:14 +08:00
|
|
|
"include",
|
|
|
|
],
|
2021-03-04 08:10:20 +08:00
|
|
|
deps = [":hlo_ops_td_files"],
|
2020-12-17 08:22:14 +08:00
|
|
|
)
|
|
|
|
|
2021-05-13 01:58:02 +08:00
|
|
|
gentbl_cc_library(
|
2020-12-17 08:22:14 +08:00
|
|
|
name = "hlo_ops_inc_gen",
|
|
|
|
strip_include_prefix = "include",
|
|
|
|
tbl_outs = [
|
2021-05-13 01:58:02 +08:00
|
|
|
(
|
|
|
|
["-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",
|
|
|
|
),
|
2020-12-17 08:22:14 +08:00
|
|
|
],
|
|
|
|
tblgen = "@llvm-project//mlir:mlir-tblgen",
|
|
|
|
td_file = "include/mlir-hlo/Dialect/mhlo/IR/hlo_ops.td",
|
|
|
|
td_includes = [
|
2021-01-09 04:06:39 +08:00
|
|
|
"external/mlir-hlo/include",
|
2020-12-17 08:22:14 +08:00
|
|
|
"include",
|
|
|
|
],
|
2021-03-04 08:10:20 +08:00
|
|
|
deps = [":hlo_ops_td_files"],
|
2020-12-17 08:22:14 +08:00
|
|
|
)
|
|
|
|
|
2021-05-13 01:58:02 +08:00
|
|
|
gentbl_cc_library(
|
2020-12-17 08:22:14 +08:00
|
|
|
name = "hlo_ops_base_inc_gen",
|
|
|
|
strip_include_prefix = "include",
|
|
|
|
tbl_outs = [
|
2021-05-13 01:58:02 +08:00
|
|
|
(
|
|
|
|
["-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",
|
|
|
|
),
|
2020-12-17 08:22:14 +08:00
|
|
|
],
|
|
|
|
tblgen = "@llvm-project//mlir:mlir-tblgen",
|
|
|
|
td_file = "include/mlir-hlo/Dialect/mhlo/IR/hlo_ops_base.td",
|
|
|
|
td_includes = [
|
2021-01-09 04:06:39 +08:00
|
|
|
"external/mlir-hlo/include",
|
2020-12-17 08:22:14 +08:00
|
|
|
"include",
|
|
|
|
],
|
2021-03-04 08:10:20 +08:00
|
|
|
deps = [":hlo_ops_td_files"],
|
2020-12-17 08:22:14 +08:00
|
|
|
)
|
|
|
|
|
2021-05-13 01:58:02 +08:00
|
|
|
gentbl_cc_library(
|
2020-12-17 08:22:14 +08:00
|
|
|
name = "hlo_ops_base_structs_inc_gen",
|
|
|
|
tbl_outs = [
|
2021-05-13 01:58:02 +08:00
|
|
|
(
|
|
|
|
["-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",
|
|
|
|
),
|
2020-12-17 08:22:14 +08:00
|
|
|
],
|
|
|
|
tblgen = "@llvm-project//mlir:mlir-tblgen",
|
|
|
|
td_file = "include/mlir-hlo/Dialect/mhlo/IR/hlo_ops_base.td",
|
|
|
|
td_includes = [
|
2021-01-09 04:06:39 +08:00
|
|
|
"external/mlir-hlo/include",
|
2020-12-17 08:22:14 +08:00
|
|
|
"include",
|
|
|
|
],
|
2021-03-04 08:10:20 +08:00
|
|
|
deps = [":hlo_ops_td_files"],
|
2020-12-17 08:22:14 +08:00
|
|
|
)
|
|
|
|
|
2021-05-13 01:58:02 +08:00
|
|
|
gentbl_cc_library(
|
2020-12-17 08:22:14 +08:00
|
|
|
name = "hlo_ops_base_enums_inc_gen",
|
|
|
|
tbl_outs = [
|
2021-05-13 01:58:02 +08:00
|
|
|
(
|
|
|
|
["-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",
|
|
|
|
),
|
2020-12-17 08:22:14 +08:00
|
|
|
],
|
|
|
|
tblgen = "@llvm-project//mlir:mlir-tblgen",
|
|
|
|
td_file = "include/mlir-hlo/Dialect/mhlo/IR/hlo_ops_base.td",
|
|
|
|
td_includes = [
|
2021-01-09 04:06:39 +08:00
|
|
|
"external/mlir-hlo/include",
|
2020-12-17 08:22:14 +08:00
|
|
|
"include",
|
|
|
|
],
|
2021-03-04 08:10:20 +08:00
|
|
|
deps = [":hlo_ops_td_files"],
|
2020-12-17 08:22:14 +08:00
|
|
|
)
|
|
|
|
|
2021-05-13 01:58:02 +08:00
|
|
|
gentbl_cc_library(
|
2020-12-17 08:22:14 +08:00
|
|
|
name = "hlo_ops_pattern_gen",
|
|
|
|
strip_include_prefix = "lib/Dialect/mhlo/IR/",
|
|
|
|
tbl_outs = [
|
|
|
|
(
|
2021-05-13 01:58:02 +08:00
|
|
|
["-gen-rewriters"],
|
2020-12-17 08:22:14 +08:00
|
|
|
"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 = [
|
2021-01-09 04:06:39 +08:00
|
|
|
"external/mlir-hlo/include",
|
2020-12-17 08:22:14 +08:00
|
|
|
"include",
|
|
|
|
],
|
2021-03-04 08:10:20 +08:00
|
|
|
deps = [
|
2020-12-17 08:22:14 +08:00
|
|
|
":hlo_ops_td_files",
|
|
|
|
"@llvm-project//mlir:StdOpsTdFiles",
|
2021-03-10 21:43:10 +08:00
|
|
|
"@llvm-project//mlir:TensorOpsTdFiles",
|
2020-12-17 08:22:14 +08:00
|
|
|
],
|
|
|
|
)
|
|
|
|
|
2021-05-13 01:58:02 +08:00
|
|
|
gentbl_cc_library(
|
2021-02-23 00:41:59 +08:00
|
|
|
name = "lhlo_ops_structs_inc_gen",
|
|
|
|
strip_include_prefix = "include",
|
|
|
|
tbl_outs = [
|
2021-05-13 01:58:02 +08:00
|
|
|
(
|
|
|
|
["-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",
|
|
|
|
),
|
2021-02-23 00:41:59 +08:00
|
|
|
],
|
|
|
|
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",
|
|
|
|
],
|
2021-03-04 08:10:20 +08:00
|
|
|
deps = [":hlo_ops_td_files"],
|
2021-02-23 00:41:59 +08:00
|
|
|
)
|
|
|
|
|
2021-05-13 01:58:02 +08:00
|
|
|
gentbl_cc_library(
|
2020-12-17 08:22:14 +08:00
|
|
|
name = "lhlo_ops_inc_gen",
|
|
|
|
strip_include_prefix = "include",
|
|
|
|
tbl_outs = [
|
2021-05-13 01:58:02 +08:00
|
|
|
(
|
|
|
|
["-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",
|
|
|
|
),
|
2020-12-17 08:22:14 +08:00
|
|
|
],
|
|
|
|
tblgen = "@llvm-project//mlir:mlir-tblgen",
|
|
|
|
td_file = "include/mlir-hlo/Dialect/mhlo/IR/lhlo_ops.td",
|
|
|
|
td_includes = [
|
2021-01-09 04:06:39 +08:00
|
|
|
"external/mlir-hlo/include",
|
2020-12-17 08:22:14 +08:00
|
|
|
"include",
|
|
|
|
],
|
2021-03-04 08:10:20 +08:00
|
|
|
deps = [":hlo_ops_td_files"],
|
2020-12-17 08:22:14 +08:00
|
|
|
)
|
|
|
|
|
2021-05-13 01:58:02 +08:00
|
|
|
gentbl_cc_library(
|
2020-12-17 08:22:14 +08:00
|
|
|
name = "lhlo_gpu_ops_structs_inc_gen",
|
|
|
|
strip_include_prefix = "include",
|
|
|
|
tbl_outs = [
|
2021-05-13 01:58:02 +08:00
|
|
|
(
|
|
|
|
["-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",
|
|
|
|
),
|
2020-12-17 08:22:14 +08:00
|
|
|
],
|
|
|
|
tblgen = "@llvm-project//mlir:mlir-tblgen",
|
|
|
|
td_file = "include/mlir-hlo/Dialect/mhlo/IR/lhlo_gpu_ops_structs.td",
|
|
|
|
td_includes = [
|
2021-01-09 04:06:39 +08:00
|
|
|
"external/mlir-hlo/include",
|
2020-12-17 08:22:14 +08:00
|
|
|
"include",
|
|
|
|
],
|
2021-03-04 08:10:20 +08:00
|
|
|
deps = [":hlo_ops_td_files"],
|
2020-12-17 08:22:14 +08:00
|
|
|
)
|
|
|
|
|
2021-05-13 01:58:02 +08:00
|
|
|
gentbl_cc_library(
|
2020-12-17 08:22:14 +08:00
|
|
|
name = "lhlo_gpu_ops_enums_inc_gen",
|
|
|
|
strip_include_prefix = "include",
|
|
|
|
tbl_outs = [
|
2021-05-13 01:58:02 +08:00
|
|
|
(
|
|
|
|
["-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",
|
|
|
|
),
|
2020-12-17 08:22:14 +08:00
|
|
|
],
|
|
|
|
tblgen = "@llvm-project//mlir:mlir-tblgen",
|
|
|
|
td_file = "include/mlir-hlo/Dialect/mhlo/IR/lhlo_gpu_ops_enums.td",
|
|
|
|
td_includes = [
|
2021-01-09 04:06:39 +08:00
|
|
|
"external/mlir-hlo/include",
|
2020-12-17 08:22:14 +08:00
|
|
|
"include",
|
|
|
|
],
|
2021-03-04 08:10:20 +08:00
|
|
|
deps = [":hlo_ops_td_files"],
|
2020-12-17 08:22:14 +08:00
|
|
|
)
|
|
|
|
|
2021-05-12 23:22:41 +08:00
|
|
|
gentbl_filegroup(
|
|
|
|
name = "hlo_ops_doc_gen",
|
|
|
|
tbl_outs = [
|
2021-05-13 01:58:02 +08:00
|
|
|
(
|
|
|
|
["-gen-dialect-doc"],
|
|
|
|
"g3doc/hlo_ops.md",
|
|
|
|
),
|
2021-05-12 23:22:41 +08:00
|
|
|
],
|
|
|
|
tblgen = "@llvm-project//mlir:mlir-tblgen",
|
|
|
|
td_file = "include/mlir-hlo/Dialect/mhlo/IR/hlo_ops.td",
|
|
|
|
deps = [":hlo_ops_td_files"],
|
|
|
|
)
|
|
|
|
|
|
|
|
gentbl_filegroup(
|
|
|
|
name = "lhlo_ops_doc_gen",
|
|
|
|
tbl_outs = [
|
2021-05-13 01:58:02 +08:00
|
|
|
(
|
|
|
|
["-gen-dialect-doc"],
|
|
|
|
"g3doc/lhlo_ops.md",
|
|
|
|
),
|
2021-05-12 23:22:41 +08:00
|
|
|
],
|
|
|
|
tblgen = "@llvm-project//mlir:mlir-tblgen",
|
|
|
|
td_file = "include/mlir-hlo/Dialect/mhlo/IR/lhlo_ops.td",
|
|
|
|
deps = [":hlo_ops_td_files"],
|
|
|
|
)
|
|
|
|
|
2021-03-11 07:36:22 +08:00
|
|
|
cc_library(
|
|
|
|
name = "hlo_ops_common",
|
|
|
|
srcs = ["lib/Dialect/mhlo/IR/hlo_ops_common.cc"],
|
|
|
|
hdrs = ["include/mlir-hlo/Dialect/mhlo/IR/hlo_ops_common.h"],
|
|
|
|
includes = ["include"],
|
|
|
|
deps = [
|
2021-05-15 00:46:42 +08:00
|
|
|
"@llvm-project//llvm:Support",
|
2021-03-11 07:36:22 +08:00
|
|
|
"@llvm-project//mlir:IR",
|
|
|
|
"@llvm-project//mlir:Support",
|
|
|
|
],
|
|
|
|
)
|
|
|
|
|
2020-12-17 08:22:14 +08:00
|
|
|
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",
|
|
|
|
],
|
|
|
|
)
|
|
|
|
|
2021-05-13 01:58:02 +08:00
|
|
|
gentbl_cc_library(
|
2020-12-17 08:22:14 +08:00
|
|
|
name = "lhlo_gpu_ops_inc_gen",
|
|
|
|
strip_include_prefix = "include",
|
|
|
|
tbl_outs = [
|
2021-05-13 01:58:02 +08:00
|
|
|
(
|
|
|
|
["-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",
|
|
|
|
),
|
2020-12-17 08:22:14 +08:00
|
|
|
],
|
|
|
|
tblgen = "@llvm-project//mlir:mlir-tblgen",
|
|
|
|
td_file = "include/mlir-hlo/Dialect/mhlo/IR/lhlo_gpu_ops.td",
|
|
|
|
td_includes = [
|
2021-01-09 04:06:39 +08:00
|
|
|
"external/mlir-hlo/include",
|
2020-12-17 08:22:14 +08:00
|
|
|
"include",
|
|
|
|
],
|
2021-03-04 08:10:20 +08:00
|
|
|
deps = [":hlo_ops_td_files"],
|
2020-12-17 08:22:14 +08:00
|
|
|
)
|
|
|
|
|
|
|
|
#TODO(aminim): revisit the naming and grouping of these rules post-move.
|
2021-05-13 01:58:02 +08:00
|
|
|
gentbl_cc_library(
|
2020-12-17 08:22:14 +08:00
|
|
|
name = "canonicalize_inc_gen",
|
|
|
|
strip_include_prefix = "lib/Dialect/mhlo/IR/",
|
|
|
|
tbl_outs = [
|
2021-05-13 01:58:02 +08:00
|
|
|
(
|
|
|
|
["-gen-rewriters"],
|
|
|
|
"lib/Dialect/mhlo/IR/mhlo_canonicalize.inc",
|
|
|
|
),
|
2020-12-17 08:22:14 +08:00
|
|
|
],
|
|
|
|
tblgen = "@llvm-project//mlir:mlir-tblgen",
|
|
|
|
td_file = "lib/Dialect/mhlo/IR/mhlo_canonicalize.td",
|
|
|
|
td_includes = [
|
2021-01-09 04:06:39 +08:00
|
|
|
"external/mlir-hlo/include",
|
2020-12-17 08:22:14 +08:00
|
|
|
"include",
|
|
|
|
],
|
2021-03-04 08:10:20 +08:00
|
|
|
deps = [":hlo_ops_td_files"],
|
2020-12-17 08:22:14 +08:00
|
|
|
)
|
|
|
|
|
2021-05-13 01:58:02 +08:00
|
|
|
gentbl_cc_library(
|
2020-12-17 08:22:14 +08:00
|
|
|
name = "infer_fusibility_op_interface_gen",
|
|
|
|
tbl_outs = [
|
|
|
|
(
|
2021-05-13 01:58:02 +08:00
|
|
|
["-gen-op-interface-decls"],
|
2020-12-17 08:22:14 +08:00
|
|
|
"include/mlir-hlo/Dialect/mhlo/IR/infer_fusibility_op_interface.h.inc",
|
|
|
|
),
|
|
|
|
(
|
2021-05-13 01:58:02 +08:00
|
|
|
["-gen-op-interface-defs"],
|
2020-12-17 08:22:14 +08:00
|
|
|
"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 = [
|
2021-01-09 04:06:39 +08:00
|
|
|
"external/mlir-hlo/include",
|
2020-12-17 08:22:14 +08:00
|
|
|
"include",
|
|
|
|
],
|
2021-03-04 08:10:20 +08:00
|
|
|
deps = [":hlo_ops_td_files"],
|
2020-12-17 08:22:14 +08:00
|
|
|
)
|
|
|
|
|
|
|
|
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 = [
|
|
|
|
":canonicalize_inc_gen",
|
|
|
|
":chlo_ops_inc_gen",
|
|
|
|
":convert_op_folder",
|
|
|
|
":hlo_ops_base_enums",
|
|
|
|
":hlo_ops_base_inc_gen",
|
|
|
|
":hlo_ops_base_structs",
|
2021-03-11 07:36:22 +08:00
|
|
|
":hlo_ops_common",
|
2020-12-17 08:22:14 +08:00
|
|
|
":hlo_ops_inc_gen",
|
2021-03-11 07:36:22 +08:00
|
|
|
":hlo_ops_pattern_gen",
|
2020-12-17 08:22:14 +08:00
|
|
|
":infer_fusibility_op_interface",
|
|
|
|
"@llvm-project//llvm:Support",
|
|
|
|
"@llvm-project//mlir:Analysis",
|
2021-05-11 20:17:01 +08:00
|
|
|
"@llvm-project//mlir:ControlFlowInterfaces",
|
2020-12-17 08:22:14 +08:00
|
|
|
"@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",
|
2020-12-24 15:53:08 +08:00
|
|
|
"@llvm-project//mlir:TensorDialect",
|
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",
|
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",
|
2020-12-17 08:22:14 +08:00
|
|
|
"lib/Dialect/mhlo/IR/lhlo_ops.cc",
|
2021-02-23 00:41:59 +08:00
|
|
|
"lib/Dialect/mhlo/IR/lhlo_ops_structs.cc",
|
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",
|
2021-03-11 07:36:22 +08:00
|
|
|
":hlo_ops_common",
|
2020-12-17 08:22:14 +08:00
|
|
|
":lhlo_ops_inc_gen",
|
2021-02-25 19:50:11 +08:00
|
|
|
":lhlo_ops_structs_inc_gen",
|
2020-12-17 08:22:14 +08:00
|
|
|
"@llvm-project//llvm:Support",
|
|
|
|
"@llvm-project//mlir:Analysis",
|
2021-05-05 15:26:46 +08:00
|
|
|
"@llvm-project//mlir:ControlFlowInterfaces",
|
2020-12-17 08:22:14 +08:00
|
|
|
"@llvm-project//mlir:CopyOpInterface",
|
|
|
|
"@llvm-project//mlir:IR",
|
2021-05-05 15:26:46 +08:00
|
|
|
"@llvm-project//mlir:LoopLikeInterface",
|
2021-03-17 04:31:59 +08:00
|
|
|
"@llvm-project//mlir:MemRefDialect",
|
2020-12-17 08:22:14 +08:00
|
|
|
"@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",
|
2021-05-21 00:40:28 +08:00
|
|
|
":hlo_ops_common",
|
2020-12-17 08:22:14 +08:00
|
|
|
":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",
|
2021-05-05 15:26:46 +08:00
|
|
|
"@llvm-project//mlir:ControlFlowInterfaces",
|
2020-12-17 08:22:14 +08:00
|
|
|
"@llvm-project//mlir:CopyOpInterface",
|
|
|
|
"@llvm-project//mlir:IR",
|
|
|
|
"@llvm-project//mlir:InferTypeOpInterface",
|
2021-05-05 15:26:46 +08:00
|
|
|
"@llvm-project//mlir:LoopLikeInterface",
|
2020-12-17 08:22:14 +08:00
|
|
|
"@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",
|
2020-12-17 12:29:15 +08:00
|
|
|
"@llvm-project//mlir:TensorDialect",
|
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",
|
2021-01-21 17:21:23 +08:00
|
|
|
"@llvm-project//mlir:ComplexDialect",
|
2020-12-17 08:22:14 +08:00
|
|
|
"@llvm-project//mlir:IR",
|
2021-02-13 00:30:51 +08:00
|
|
|
"@llvm-project//mlir:MathDialect",
|
2020-12-22 07:26:38 +08:00
|
|
|
"@llvm-project//mlir:SCFDialect",
|
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",
|
2021-03-17 04:31:59 +08:00
|
|
|
"@llvm-project//mlir:MemRefDialect",
|
2020-12-17 08:22:14 +08:00
|
|
|
"@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",
|
2021-02-13 00:30:51 +08:00
|
|
|
"@llvm-project//mlir:MathDialect",
|
2021-03-30 16:06:12 +08:00
|
|
|
"@llvm-project//mlir:MemRefDialect",
|
2020-12-17 08:22:14 +08:00
|
|
|
"@llvm-project//mlir:Pass",
|
2020-12-22 07:26:38 +08:00
|
|
|
"@llvm-project//mlir:SCFDialect",
|
2020-12-17 08:22:14 +08:00
|
|
|
"@llvm-project//mlir:StandardOps",
|
2021-02-25 01:43:54 +08:00
|
|
|
"@llvm-project//mlir:Support",
|
2021-01-12 02:11:39 +08:00
|
|
|
"@llvm-project//mlir:TensorDialect",
|
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",
|
2020-12-24 15:53:08 +08:00
|
|
|
"@llvm-project//mlir:TensorDialect",
|
2020-12-17 08:22:14 +08:00
|
|
|
"@llvm-project//mlir:Transforms",
|
|
|
|
],
|
|
|
|
alwayslink = 1,
|
|
|
|
)
|
|
|
|
|
2021-03-10 22:20:43 +08:00
|
|
|
cc_library(
|
|
|
|
name = "move_up_dynamic_broadcasts_for_fusion",
|
|
|
|
srcs = ["lib/Dialect/mhlo/transforms/move_up_dynamic_broadcasts_for_fusion.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:InferTypeOpInterface",
|
|
|
|
"@llvm-project//mlir:Pass",
|
|
|
|
"@llvm-project//mlir:SCFDialect",
|
|
|
|
"@llvm-project//mlir:Shape",
|
|
|
|
"@llvm-project//mlir:StandardOps",
|
|
|
|
"@llvm-project//mlir:TensorDialect",
|
|
|
|
"@llvm-project//mlir:Transforms",
|
|
|
|
],
|
|
|
|
alwayslink = 1,
|
|
|
|
)
|
|
|
|
|
2021-05-12 18:45:09 +08:00
|
|
|
cc_library(
|
|
|
|
name = "rank_specialization",
|
|
|
|
srcs = ["lib/Dialect/mhlo/transforms/rank_specialization.cc"],
|
|
|
|
hdrs = [
|
|
|
|
"include/mlir-hlo/Dialect/mhlo/transforms/passes.h",
|
|
|
|
"include/mlir-hlo/Dialect/mhlo/transforms/rewriters.h",
|
|
|
|
],
|
|
|
|
deps = [
|
|
|
|
":hlo",
|
2021-06-01 21:53:30 +08:00
|
|
|
":pass_details",
|
2021-05-12 18:45:09 +08:00
|
|
|
"@llvm-project//llvm:Support",
|
|
|
|
"@llvm-project//mlir:IR",
|
|
|
|
"@llvm-project//mlir:InferTypeOpInterface",
|
|
|
|
"@llvm-project//mlir:Pass",
|
|
|
|
"@llvm-project//mlir:SCFDialect",
|
|
|
|
"@llvm-project//mlir:Shape",
|
|
|
|
"@llvm-project//mlir:StandardOps",
|
|
|
|
"@llvm-project//mlir:TensorDialect",
|
|
|
|
"@llvm-project//mlir:Transforms",
|
|
|
|
],
|
|
|
|
alwayslink = 1,
|
|
|
|
)
|
|
|
|
|
2020-12-17 08:22:14 +08:00
|
|
|
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",
|
2021-03-30 16:06:12 +08:00
|
|
|
"@llvm-project//mlir:MemRefDialect",
|
2020-12-17 08:22:14 +08:00
|
|
|
"@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",
|
2021-03-17 04:31:59 +08:00
|
|
|
"@llvm-project//mlir:MemRefDialect",
|
2020-12-17 08:22:14 +08:00
|
|
|
"@llvm-project//mlir:Pass",
|
|
|
|
"@llvm-project//mlir:SCFDialect",
|
|
|
|
"@llvm-project//mlir:StandardOps",
|
|
|
|
"@llvm-project//mlir:Support",
|
2020-12-24 15:53:08 +08:00
|
|
|
"@llvm-project//mlir:TensorDialect",
|
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",
|
2021-03-30 16:06:12 +08:00
|
|
|
"@llvm-project//mlir:MemRefDialect",
|
2020-12-17 08:22:14 +08:00
|
|
|
"@llvm-project//mlir:Pass",
|
|
|
|
"@llvm-project//mlir:Shape",
|
|
|
|
"@llvm-project//mlir:ShapeTransforms",
|
|
|
|
"@llvm-project//mlir:StandardOps",
|
|
|
|
"@llvm-project//mlir:StandardOpsTransforms",
|
|
|
|
"@llvm-project//mlir:Support",
|
2020-12-17 12:29:15 +08:00
|
|
|
"@llvm-project//mlir:TensorDialect",
|
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,
|
|
|
|
)
|
|
|
|
|
2021-05-13 01:58:02 +08:00
|
|
|
gentbl_cc_library(
|
2020-12-17 08:22:14 +08:00
|
|
|
name = "legalize_to_standard_inc_gen",
|
|
|
|
strip_include_prefix = "lib/Dialect/mhlo/transforms/",
|
|
|
|
tbl_outs = [
|
2021-05-13 01:58:02 +08:00
|
|
|
(
|
|
|
|
["-gen-rewriters"],
|
|
|
|
"lib/Dialect/mhlo/transforms/generated_legalize_to_standard.inc",
|
|
|
|
),
|
2020-12-17 08:22:14 +08:00
|
|
|
],
|
|
|
|
tblgen = "@llvm-project//mlir:mlir-tblgen",
|
|
|
|
td_file = "lib/Dialect/mhlo/transforms/legalize_to_standard_patterns.td",
|
|
|
|
td_includes = [
|
2021-01-09 04:06:39 +08:00
|
|
|
"external/mlir-hlo/include",
|
2020-12-17 08:22:14 +08:00
|
|
|
"include",
|
|
|
|
],
|
2021-03-04 08:10:20 +08:00
|
|
|
deps = [
|
2020-12-17 08:22:14 +08:00
|
|
|
":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",
|
2020-12-17 12:29:15 +08:00
|
|
|
"@llvm-project//mlir:TensorDialect",
|
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",
|
2021-02-13 00:30:51 +08:00
|
|
|
"@llvm-project//mlir:MathDialect",
|
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,
|
|
|
|
)
|
|
|
|
|
2021-05-13 01:58:02 +08:00
|
|
|
gentbl_cc_library(
|
2020-12-17 08:22:14 +08:00
|
|
|
name = "lower_complex_inc_gen",
|
|
|
|
strip_include_prefix = "lib/Dialect/mhlo/transforms/",
|
|
|
|
tbl_outs = [
|
2021-05-13 01:58:02 +08:00
|
|
|
(
|
|
|
|
["-gen-rewriters"],
|
|
|
|
"lib/Dialect/mhlo/transforms/generated_lower_complex.inc",
|
|
|
|
),
|
2020-12-17 08:22:14 +08:00
|
|
|
],
|
|
|
|
tblgen = "@llvm-project//mlir:mlir-tblgen",
|
|
|
|
td_file = "lib/Dialect/mhlo/transforms/lower_complex_patterns.td",
|
|
|
|
td_includes = [
|
2021-01-09 04:06:39 +08:00
|
|
|
"external/mlir-hlo/include",
|
2020-12-17 08:22:14 +08:00
|
|
|
"include",
|
|
|
|
],
|
2021-03-04 08:10:20 +08:00
|
|
|
deps = [
|
2020-12-17 08:22:14 +08:00
|
|
|
":hlo_ops_td_files",
|
|
|
|
"@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",
|
2021-03-17 04:31:59 +08:00
|
|
|
"@llvm-project//mlir:MemRefDialect",
|
2020-12-17 08:22:14 +08:00
|
|
|
"@llvm-project//mlir:StandardOps",
|
2021-01-20 23:08:32 +08:00
|
|
|
"@llvm-project//mlir:TensorDialect",
|
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",
|
2021-02-17 01:46:33 +08:00
|
|
|
"@llvm-project//llvm:Support",
|
2020-12-17 08:22:14 +08:00
|
|
|
"@llvm-project//mlir:IR",
|
|
|
|
"@llvm-project//mlir:SCFDialect",
|
|
|
|
"@llvm-project//mlir:Shape",
|
|
|
|
"@llvm-project//mlir:StandardOps",
|
2020-12-24 15:53:08 +08:00
|
|
|
"@llvm-project//mlir:TensorDialect",
|
2020-12-17 08:22:14 +08:00
|
|
|
"@llvm-project//mlir:Transforms",
|
|
|
|
],
|
|
|
|
)
|
|
|
|
|
2021-05-13 01:58:02 +08:00
|
|
|
gentbl_cc_library(
|
2020-12-17 08:22:14 +08:00
|
|
|
name = "chlo_legalize_to_hlo_inc_gen",
|
|
|
|
strip_include_prefix = "lib/Dialect/mhlo/transforms/",
|
|
|
|
tbl_outs = [
|
|
|
|
(
|
2021-05-13 01:58:02 +08:00
|
|
|
["-gen-rewriters"],
|
2020-12-17 08:22:14 +08:00
|
|
|
"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 = [
|
2021-01-09 04:06:39 +08:00
|
|
|
"external/mlir-hlo/include",
|
2020-12-17 08:22:14 +08:00
|
|
|
"include",
|
|
|
|
],
|
2021-03-04 08:10:20 +08:00
|
|
|
deps = [":hlo_ops_td_files"],
|
2020-12-17 08:22:14 +08:00
|
|
|
)
|
|
|
|
|
|
|
|
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",
|
2021-03-30 16:06:12 +08:00
|
|
|
"@llvm-project//mlir:MemRefDialect",
|
2020-12-17 08:22:14 +08:00
|
|
|
"@llvm-project//mlir:Pass",
|
|
|
|
"@llvm-project//mlir:SCFDialect",
|
|
|
|
"@llvm-project//mlir:Shape",
|
|
|
|
"@llvm-project//mlir:StandardOps",
|
2020-12-24 15:53:08 +08:00
|
|
|
"@llvm-project//mlir:TensorDialect",
|
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",
|
2021-03-10 22:20:43 +08:00
|
|
|
":move_up_dynamic_broadcasts_for_fusion",
|
2021-05-12 18:45:09 +08:00
|
|
|
":rank_specialization",
|
2020-12-17 08:22:14 +08:00
|
|
|
":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",
|
2021-04-03 05:25:23 +08:00
|
|
|
"@llvm-project//mlir:AllPassesAndDialects",
|
2020-12-17 08:22:14 +08:00
|
|
|
"@llvm-project//mlir:IR",
|
|
|
|
"@llvm-project//mlir:MlirOptLib",
|
|
|
|
"@llvm-project//mlir:Pass",
|
|
|
|
"@llvm-project//mlir:Support",
|
|
|
|
],
|
|
|
|
)
|