STT-tensorflow/third_party/mlir/BUILD

3635 lines
86 KiB
Python

# Description:
# The MLIR "Multi-Level Intermediate Representation" Compiler Infrastructure
load("@org_tensorflow//third_party/mlir:tblgen.bzl", "gentbl")
load("@org_tensorflow//third_party/mlir:linalggen.bzl", "genlinalg")
licenses(["notice"])
package(default_visibility = [":friends"])
package_group(
name = "subpackages",
packages = ["//..."],
)
package_group(
name = "friends",
packages = ["//..."],
)
exports_files([
"LICENSE.TXT",
"include/mlir/Dialect/LLVMIR/LLVMOps.td",
"run_lit.sh",
])
cc_library(
name = "DialectSymbolRegistry",
# strip_include_prefix does not apply to textual_hdrs.
hdrs = ["include/mlir/IR/DialectSymbolRegistry.def"],
strip_include_prefix = "include/mlir/IR",
textual_hdrs = ["include/mlir/IR/DialectSymbolRegistry.def"],
)
gentbl(
name = "OpAsmInterfacesIncGen",
strip_include_prefix = "include",
tbl_outs = [
(
"-gen-op-interface-decls",
"include/mlir/IR/OpAsmInterface.h.inc",
),
(
"-gen-op-interface-defs",
"include/mlir/IR/OpAsmInterface.cpp.inc",
),
],
tblgen = ":mlir-tblgen",
td_file = "include/mlir/IR/OpAsmInterface.td",
td_srcs = [
":OpBaseTdFiles",
],
)
gentbl(
name = "SymbolInterfacesIncGen",
strip_include_prefix = "include",
tbl_outs = [
(
"-gen-op-interface-decls",
"include/mlir/IR/SymbolInterfaces.h.inc",
),
(
"-gen-op-interface-defs",
"include/mlir/IR/SymbolInterfaces.cpp.inc",
),
],
tblgen = ":mlir-tblgen",
td_file = "include/mlir/IR/SymbolInterfaces.td",
td_srcs = [
":OpBaseTdFiles",
],
)
cc_library(
name = "IR",
srcs = glob([
"lib/IR/*.cpp",
"lib/IR/*.h",
]),
hdrs = glob([
"include/mlir/IR/*.h",
]) + [
"include/mlir/Interfaces/CallInterfaces.h",
],
includes = ["include"],
deps = [
":CallOpInterfacesIncGen",
":DialectSymbolRegistry",
":InferTypeOpInterfaceIncGen",
":OpAsmInterfacesIncGen",
":SideEffectInterfacesIncGen",
":Support",
":SymbolInterfacesIncGen",
"@llvm-project//llvm:Support",
],
)
cc_library(
name = "Pass",
srcs = glob([
"lib/Pass/*.cpp",
"lib/Pass/*.h",
]),
hdrs = glob([
"include/mlir/Pass/*.h",
]),
includes = ["include"],
linkopts = [
"-lm",
"-lpthread",
],
deps = [
":Analysis",
":IR",
":Support",
"@llvm-project//llvm:Support",
],
)
# TODO(ntv): Update these to enable simplifying the cmake and build files.
cc_library(
name = "EDSC",
srcs = [
"lib/EDSC/Builders.cpp",
],
hdrs = [
"include/mlir-c/Core.h",
"include/mlir/EDSC/Builders.h",
],
includes = ["include"],
deps = [
":IR",
":Support",
"@llvm-project//llvm:Support",
],
)
cc_library(
name = "EDSCInterface",
srcs = [
"lib/EDSC/CoreAPIs.cpp",
],
hdrs = [
"include/mlir-c/Core.h",
],
includes = ["include"],
deps = [
":IR",
":Parser",
":Support",
"@llvm-project//llvm:Support",
],
)
filegroup(
name = "OpBaseTdFiles",
srcs = [
"include/mlir/Dialect/Affine/IR/AffineOpsBase.td",
"include/mlir/IR/OpBase.td",
],
)
##---------------------------------------------------------------------------##
# Affine dialect.
##---------------------------------------------------------------------------##
filegroup(
name = "PassBaseTdFiles",
srcs = [
"include/mlir/Pass/PassBase.td",
],
)
filegroup(
name = "AffineOpsTdFiles",
srcs = [
"include/mlir/Dialect/Affine/IR/AffineMemoryOpInterfaces.td",
"include/mlir/Dialect/Affine/IR/AffineOps.td",
"include/mlir/Dialect/Affine/IR/AffineOpsBase.td",
"include/mlir/Interfaces/LoopLikeInterface.td",
"include/mlir/Interfaces/SideEffectInterfaces.td",
":OpBaseTdFiles",
],
)
gentbl(
name = "AffineOpsIncGen",
strip_include_prefix = "include",
tbl_outs = [
(
"-gen-op-decls",
"include/mlir/Dialect/Affine/IR/AffineOps.h.inc",
),
(
"-gen-op-defs",
"include/mlir/Dialect/Affine/IR/AffineOps.cpp.inc",
),
(
"-gen-dialect-decls",
"include/mlir/Dialect/Affine/IR/AffineOpsDialect.h.inc",
),
],
tblgen = ":mlir-tblgen",
td_file = "include/mlir/Dialect/Affine/IR/AffineOps.td",
td_srcs = [
":AffineOpsTdFiles",
],
)
gentbl(
name = "AffineMemoryOpInterfacesIncGen",
strip_include_prefix = "include",
tbl_outs = [
(
"-gen-op-interface-decls",
"include/mlir/Dialect/Affine/IR/AffineMemoryOpInterfaces.h.inc",
),
(
"-gen-op-interface-defs",
"include/mlir/Dialect/Affine/IR/AffineMemoryOpInterfaces.cpp.inc",
),
],
tblgen = ":mlir-tblgen",
td_file = "include/mlir/Dialect/Affine/IR/AffineMemoryOpInterfaces.td",
td_srcs = [
":AffineOpsTdFiles",
],
)
##---------------------------------------------------------------------------##
# AVX512 dialect.
##---------------------------------------------------------------------------##
filegroup(
name = "AVX512TdFiles",
srcs = [
"include/mlir/Dialect/AVX512/AVX512.td",
"include/mlir/Dialect/LLVMIR/LLVMOpBase.td",
"include/mlir/IR/OpBase.td",
"include/mlir/Interfaces/SideEffectInterfaces.td",
],
)
gentbl(
name = "AVX512IncGen",
strip_include_prefix = "include",
tbl_outs = [
(
"-gen-dialect-decls -dialect=avx512",
"include/mlir/Dialect/AVX512/AVX512Dialect.h.inc",
),
(
"-gen-op-decls",
"include/mlir/Dialect/AVX512/AVX512.h.inc",
),
(
"-gen-op-defs",
"include/mlir/Dialect/AVX512/AVX512.cpp.inc",
),
(
"-gen-op-doc",
"g3doc/Dialects/AVX512/AVX512.md",
),
],
tblgen = ":mlir-tblgen",
td_file = "include/mlir/Dialect/AVX512/AVX512.td",
td_srcs = [
":AVX512TdFiles",
],
)
cc_library(
name = "AVX512",
srcs = [
"lib/Dialect/AVX512/IR/AVX512Dialect.cpp",
],
hdrs = [
"include/mlir/Dialect/AVX512/AVX512Dialect.h",
],
includes = ["include"],
deps = [
":AVX512IncGen",
":IR",
":SideEffectInterfaces",
":VectorOps",
"@llvm-project//llvm:Core",
"@llvm-project//llvm:Support",
],
)
cc_library(
name = "AVX512ToLLVM",
srcs = glob([
"lib/Conversion/AVX512ToLLVM/*.cpp",
]) + ["lib/Conversion/PassDetail.h"],
hdrs = glob([
"include/mlir/Conversion/AVX512ToLLVM/*.h",
]),
includes = ["include"],
deps = [
":AVX512",
":ConversionPassIncGen",
":EDSC",
":IR",
":LLVMAVX512",
":LLVMDialect",
":Pass",
":StandardOps",
":StandardToLLVM",
":Support",
":Transforms",
":VectorOps",
":VectorToLLVM",
":VectorToSCF",
"@llvm-project//llvm:Core",
"@llvm-project//llvm:Support",
],
)
filegroup(
name = "SCFTdFiles",
srcs = [
"include/mlir/Dialect/SCF/SCFOps.td",
"include/mlir/Interfaces/ControlFlowInterfaces.td",
"include/mlir/Interfaces/LoopLikeInterface.td",
"include/mlir/Interfaces/SideEffectInterfaces.td",
":OpBaseTdFiles",
],
)
gentbl(
name = "SCFIncGen",
strip_include_prefix = "include",
tbl_outs = [
(
"-gen-op-decls",
"include/mlir/Dialect/SCF/SCFOps.h.inc",
),
(
"-gen-op-defs",
"include/mlir/Dialect/SCF/SCFOps.cpp.inc",
),
(
"-gen-dialect-decls",
"include/mlir/Dialect/SCF/SCFOpsDialect.h.inc",
),
],
tblgen = ":mlir-tblgen",
td_file = "include/mlir/Dialect/SCF/SCFOps.td",
td_srcs = [
":SCFTdFiles",
],
)
gentbl(
name = "LoopPassIncGen",
strip_include_prefix = "include",
tbl_outs = [
(
"-gen-pass-decls",
"include/mlir/Dialect/SCF/Passes.h.inc",
),
],
tblgen = ":mlir-tblgen",
td_file = "include/mlir/Dialect/SCF/Passes.td",
td_srcs = [
":PassBaseTdFiles",
],
)
cc_library(
name = "SCFTransforms",
srcs = glob([
"lib/Dialect/SCF/Transforms/*.cpp",
"lib/Dialect/SCF/Transforms/*.h",
]),
hdrs = ["include/mlir/Dialect/SCF/Passes.h"],
includes = ["include"],
deps = [
":Affine",
":IR",
":LoopPassIncGen",
":Pass",
":SCFDialect",
":StandardOps",
":Transforms",
"@llvm-project//llvm:Support",
],
)
filegroup(
name = "StdOpsTdFiles",
srcs = [
"include/mlir/Dialect/StandardOps/IR/Ops.td",
"include/mlir/IR/OpAsmInterface.td",
"include/mlir/Interfaces/CallInterfaces.td",
"include/mlir/Interfaces/ControlFlowInterfaces.td",
"include/mlir/Interfaces/SideEffectInterfaces.td",
"include/mlir/Interfaces/ViewLikeInterface.td",
":OpBaseTdFiles",
],
)
gentbl(
name = "StandardOpsIncGen",
strip_include_prefix = "include",
tbl_outs = [
(
"-gen-op-decls",
"include/mlir/Dialect/StandardOps/IR/Ops.h.inc",
),
(
"-gen-op-defs",
"include/mlir/Dialect/StandardOps/IR/Ops.cpp.inc",
),
(
"-gen-dialect-decls",
"include/mlir/Dialect/StandardOps/IR/OpsDialect.h.inc",
),
(
"-gen-enum-decls",
"include/mlir/Dialect/StandardOps/IR/OpsEnums.h.inc",
),
(
"-gen-enum-defs",
"include/mlir/Dialect/StandardOps/IR/OpsEnums.cpp.inc",
),
],
tblgen = ":mlir-tblgen",
td_file = "include/mlir/Dialect/StandardOps/IR/Ops.td",
td_srcs = [
":StdOpsTdFiles",
],
)
cc_library(
name = "Dialect",
srcs = glob([
"lib/Dialect/*.cpp",
"lib/Dialect/*.h",
]),
hdrs = glob([
"include/mlir/Dialect/*.h",
]),
includes = ["include"],
deps = [
":IR",
"@llvm-project//llvm:Support",
],
)
cc_library(
name = "DialectUtils",
srcs = glob([
"lib/Dialect/Utils/*.cpp",
"lib/Dialect/Utils/*.h",
]),
hdrs = glob([
"include/mlir/Dialect/Utils/*.h",
]),
includes = ["include"],
deps = [
":IR",
":Support",
"@llvm-project//llvm:Support",
],
)
cc_library(
name = "Affine",
srcs = glob(
[
"lib/Dialect/Affine/IR/*.cpp",
"lib/Dialect/Affine/IR/*.h",
"lib/Dialect/Affine/EDSC/*.cpp",
],
) + [
"include/mlir/Transforms/InliningUtils.h",
],
hdrs = glob([
"include/mlir/Dialect/Affine/IR/*.h",
"include/mlir/Dialect/Affine/EDSC/*.h",
]),
includes = ["include"],
deps = [
":AffineMemoryOpInterfacesIncGen",
":AffineOpsIncGen",
":EDSC",
":IR",
":LoopLikeInterface",
":SideEffectInterfaces",
":StandardOps",
":Support",
"@llvm-project//llvm:Support",
],
)
cc_library(
name = "AffineUtils",
srcs = glob(
[
"lib/Dialect/Affine/Utils/*.cpp",
"lib/Dialect/Affine/Utils/*.h",
],
),
hdrs = ["include/mlir/Dialect/Affine/Utils.h"],
includes = ["include"],
deps = [
":Affine",
":IR",
"@llvm-project//llvm:Support",
],
)
gentbl(
name = "AffinePassIncGen",
strip_include_prefix = "include",
tbl_outs = [
(
"-gen-pass-decls",
"include/mlir/Dialect/Affine/Passes.h.inc",
),
],
tblgen = ":mlir-tblgen",
td_file = "include/mlir/Dialect/Affine/Passes.td",
td_srcs = [
":PassBaseTdFiles",
],
)
cc_library(
name = "AffineTransforms",
srcs = glob([
"lib/Dialect/Affine/Transforms/*.cpp",
"lib/Dialect/Affine/Transforms/*.h",
]),
hdrs = [
"include/mlir/Dialect/Affine/Passes.h",
],
includes = ["include"],
deps = [
":Affine",
":AffinePassIncGen",
":Analysis",
":IR",
":Pass",
":SCFDialect",
":StandardOps",
":Support",
":Transforms",
":VectorOps",
"@llvm-project//llvm:Support",
],
)
gentbl(
name = "ConversionPassIncGen",
strip_include_prefix = "include",
tbl_outs = [
(
"-gen-pass-decls",
"include/mlir/Conversion/Passes.h.inc",
),
],
tblgen = ":mlir-tblgen",
td_file = "include/mlir/Conversion/Passes.td",
td_srcs = [
":PassBaseTdFiles",
],
)
cc_library(
name = "AffineToStandard",
srcs = glob([
"lib/Conversion/AffineToStandard/*.cpp",
"lib/Conversion/AffineToStandard/*.h",
]) + ["lib/Conversion/PassDetail.h"],
hdrs = glob(["include/mlir/Conversion/AffineToStandard/*.h"]),
includes = ["include"],
deps = [
":Affine",
":ConversionPassIncGen",
":IR",
":Pass",
":SCFDialect",
":StandardOps",
":Support",
":Transforms",
":VectorOps",
],
)
alias(
name = "AffineToStandardTransforms",
actual = "AffineToStandard",
)
# SDBM dialect only contains attribute components that can be constructed given
# a dialect object, so whenever it is used it must also be registered. Therefore
# we don't split out the registration library for it.
cc_library(
name = "SDBM",
srcs = glob([
"lib/Dialect/SDBM/*.cpp",
"lib/Dialect/SDBM/*.h",
]),
hdrs = glob([
"include/mlir/Dialect/SDBM/*.h",
]),
includes = ["include"],
deps = [
":IR",
":Support",
"@llvm-project//llvm:Support",
],
)
cc_library(
name = "SCFDialect",
srcs = glob(
[
"lib/Dialect/SCF/*.cpp",
"lib/Dialect/SCF/*.h",
"lib/Dialect/SCF/EDSC/*.cpp",
],
),
hdrs = glob([
"include/mlir/Dialect/SCF/*.h",
"include/mlir/Dialect/SCF/EDSC/*.h",
]),
includes = ["include"],
deps = [
":ControlFlowInterfaces",
":EDSC",
":IR",
":LoopLikeInterface",
":SCFIncGen",
":SideEffectInterfaces",
":StandardOps",
":Support",
"@llvm-project//llvm:Support",
],
)
cc_library(
name = "LoopLikeInterface",
srcs = ["lib/Interfaces/LoopLikeInterface.cpp"],
hdrs = ["include/mlir/Interfaces/LoopLikeInterface.h"],
includes = ["include"],
deps = [
":IR",
":LoopLikeInterfaceIncGen",
],
)
cc_library(
name = "ViewLikeInterface",
srcs = ["lib/Interfaces/ViewLikeInterface.cpp"],
hdrs = ["include/mlir/Interfaces/ViewLikeInterface.h"],
includes = ["include"],
deps = [
":IR",
":ViewLikeInterfaceIncGen",
],
)
gentbl(
name = "ShapeOpsIncGen",
strip_include_prefix = "include",
tbl_outs = [
(
"-gen-op-decls",
"include/mlir/Dialect/Shape/IR/ShapeOps.h.inc",
),
(
"-gen-op-defs",
"include/mlir/Dialect/Shape/IR/ShapeOps.cpp.inc",
),
(
"-gen-dialect-decls",
"include/mlir/Dialect/Shape/IR/ShapeOpsDialect.h.inc",
),
],
tblgen = ":mlir-tblgen",
td_file = "include/mlir/Dialect/Shape/IR/ShapeOps.td",
td_srcs = [
":StdOpsTdFiles",
"include/mlir/Dialect/Shape/IR/ShapeBase.td",
"include/mlir/Interfaces/InferTypeOpInterface.td",
],
)
gentbl(
name = "MLIRShapeCanonicalizationIncGen",
strip_include_prefix = "include/mlir/Dialect/Shape/IR",
tbl_outs = [
(
"-gen-rewriters",
"include/mlir/Dialect/Shape/IR/ShapeCanonicalization.inc",
),
],
tblgen = ":mlir-tblgen",
td_file = "lib/Dialect/Shape/IR/ShapeCanonicalization.td",
td_srcs = [
":StdOpsTdFiles",
"include/mlir/Dialect/Shape/IR/ShapeBase.td",
"include/mlir/Dialect/Shape/IR/ShapeOps.td",
"include/mlir/Interfaces/InferTypeOpInterface.td",
],
)
cc_library(
name = "Shape",
srcs = glob(["lib/Dialect/Shape/IR/*.cpp"]),
hdrs = ["include/mlir/Dialect/Shape/IR/Shape.h"],
includes = ["include"],
deps = [
":CallOpInterfaces",
":CommonFolders",
":ControlFlowInterfaces",
":Dialect",
":IR",
":InferTypeOpInterface",
":MLIRShapeCanonicalizationIncGen",
":ShapeOpsIncGen",
":SideEffectInterfaces",
":Support",
"@llvm-project//llvm:Support",
],
)
cc_library(
name = "ShapeToStandard",
srcs = glob([
"lib/Conversion/ShapeToStandard/*.cpp",
"lib/Conversion/ShapeToStandard/*.h",
]) + ["lib/Conversion/PassDetail.h"],
hdrs = ["include/mlir/Conversion/ShapeToStandard/ShapeToStandard.h"],
includes = ["include"],
deps = [
":ConversionPassIncGen",
":Pass",
":SCFDialect",
":Shape",
":ShapeToStandardPatternsIncGen",
":StandardOps",
":Support",
":Transforms",
],
)
gentbl(
name = "ShapeToStandardPatternsIncGen",
strip_include_prefix = "include/mlir/Conversion/ShapeToStandard",
tbl_outs = [
(
"-gen-rewriters",
"include/mlir/Conversion/ShapeToStandard/ShapeToStandardPatterns.inc",
),
],
tblgen = ":mlir-tblgen",
td_file = "lib/Conversion/ShapeToStandard/ShapeToStandardPatterns.td",
td_srcs = [
":StdOpsTdFiles",
"include/mlir/Dialect/Shape/IR/ShapeBase.td",
"include/mlir/Dialect/Shape/IR/ShapeOps.td",
"include/mlir/Interfaces/InferTypeOpInterface.td",
],
)
cc_library(
name = "ShapeToSCF",
srcs = glob([
"lib/Conversion/ShapeToSCF/*.cpp",
"lib/Conversion/ShapeToSCF/*.h",
]) + ["lib/Conversion/PassDetail.h"],
hdrs = ["include/mlir/Conversion/ShapeToSCF/ShapeToSCF.h"],
includes = ["include"],
deps = [
":ConversionPassIncGen",
":IR",
":Pass",
":SCFDialect",
":Shape",
":StandardOps",
":Transforms",
],
)
gentbl(
name = "ShapeTransformsPassIncGen",
strip_include_prefix = "include",
tbl_outs = [(
"-gen-pass-decls",
"include/mlir/Dialect/Shape/Transforms/Passes.h.inc",
)],
tblgen = ":mlir-tblgen",
td_file = "include/mlir/Dialect/Shape/Transforms/Passes.td",
td_srcs = [":PassBaseTdFiles"],
)
cc_library(
name = "ShapeTransforms",
srcs = glob([
"lib/Dialect/Shape/Transforms/*.cpp",
"lib/Dialect/Shape/Transforms/*.h",
]),
hdrs = ["include/mlir/Dialect/Shape/Transforms/Passes.h"],
includes = ["include"],
deps = [
":IR",
":Pass",
":Shape",
":ShapeTransformsPassIncGen",
":Support",
":Transforms",
],
)
cc_library(
name = "StandardOps",
srcs = glob(
[
"lib/Dialect/StandardOps/IR/*.cpp",
"lib/Dialect/StandardOps/IR/*.h",
"lib/Dialect/StandardOps/EDSC/*.cpp",
],
),
hdrs = glob([
"include/mlir/Dialect/StandardOps/IR/*.h",
"include/mlir/Dialect/StandardOps/EDSC/*.h",
]) + ["include/mlir/Transforms/InliningUtils.h"],
includes = ["include"],
deps = [
":CallOpInterfaces",
":CommonFolders",
":ControlFlowInterfaces",
":EDSC",
":IR",
":SideEffectInterfaces",
":StandardOpsIncGen",
":Support",
":ViewLikeInterface",
"@llvm-project//llvm:Support",
],
)
gentbl(
name = "StandardOpsTransformsPassIncGen",
strip_include_prefix = "include",
tbl_outs = [(
"-gen-pass-decls",
"include/mlir/Dialect/StandardOps/Transforms/Passes.h.inc",
)],
tblgen = ":mlir-tblgen",
td_file = "include/mlir/Dialect/StandardOps/Transforms/Passes.td",
td_srcs = [":PassBaseTdFiles"],
)
cc_library(
name = "StandardOpsTransforms",
srcs = glob([
"lib/Dialect/StandardOps/Transforms/*.cpp",
"lib/Dialect/StandardOps/Transforms/*.h",
]),
hdrs = glob(["include/mlir/Dialect/StandardOps/Transforms/*.h"]),
includes = ["include"],
deps = [
":Analysis",
":ControlFlowInterfaces",
":IR",
":Pass",
":StandardOps",
":StandardOpsTransformsPassIncGen",
":Support",
":Transforms",
"@llvm-project//llvm:Support",
],
)
cc_library(
name = "VectorOps",
srcs = glob(
[
"lib/Dialect/Vector/*.cpp",
"lib/Dialect/Vector/*.h",
"lib/Dialect/Vector/EDSC/*.cpp",
"lib/Dialect/Vector/EDSC/*.h",
],
),
hdrs = glob([
"include/mlir/Dialect/Vector/*.h",
"include/mlir/Dialect/Vector/EDSC/*.h",
]),
includes = ["include"],
deps = [
":Affine",
":Analysis",
":DialectUtils",
":EDSC",
":IR",
":SideEffectInterfaces",
":StandardOps",
":Support",
":VectorOpsIncGen",
":VectorTransformPatternsIncGen",
"@llvm-project//llvm:Support",
],
)
cc_library(
name = "Support",
srcs = glob(
[
"lib/Support/*.cpp",
"lib/Support/*.h",
],
exclude = [
# TODO(jpienaar): Move this out, else Support depends on Analysis/
"lib/Support/MlirOptMain.cpp",
],
),
hdrs = glob([
"include/mlir/Support/*.h",
]),
includes = ["include"],
deps = [
"@llvm-project//llvm:Support",
],
)
cc_library(
name = "ParserTokenKinds",
# strip_include_prefix does not apply to textual_hdrs.
hdrs = ["lib/Parser/TokenKinds.def"],
strip_include_prefix = "lib/Parser",
textual_hdrs = ["lib/Parser/TokenKinds.def"],
)
cc_library(
name = "Parser",
srcs = glob([
"lib/Parser/*.cpp",
"lib/Parser/*.h",
]),
hdrs = [
"include/mlir/Parser.h",
],
includes = ["include"],
deps = [
":Analysis",
":IR",
":ParserTokenKinds",
":Support",
"@llvm-project//llvm:Support",
],
)
filegroup(
name = "LLVMAVX512TdFiles",
srcs = [
"include/mlir/Dialect/LLVMIR/LLVMAVX512.td",
":LLVMOpsTdFiles",
],
)
gentbl(
name = "LLVMAVX512IncGen",
strip_include_prefix = "include",
tbl_outs = [
(
"-gen-dialect-decls -dialect=llvm_avx512",
"include/mlir/Dialect/LLVMIR/LLVMAVX512Dialect.h.inc",
),
(
"-gen-op-decls",
"include/mlir/Dialect/LLVMIR/LLVMAVX512.h.inc",
),
(
"-gen-op-defs",
"include/mlir/Dialect/LLVMIR/LLVMAVX512.cpp.inc",
),
(
"-gen-op-doc",
"g3doc/Dialects/LLVMIR/LLVMAVX512.md",
),
],
tblgen = ":mlir-tblgen",
td_file = "include/mlir/Dialect/LLVMIR/LLVMAVX512.td",
td_srcs = [
":LLVMAVX512TdFiles",
],
)
cc_library(
name = "LLVMAVX512",
srcs = [
"lib/Dialect/LLVMIR/IR/LLVMAVX512Dialect.cpp",
],
hdrs = [
"include/mlir/Dialect/LLVMIR/LLVMAVX512Dialect.h",
],
includes = ["include"],
deps = [
":IR",
":LLVMAVX512IncGen",
":LLVMDialect",
"@llvm-project//llvm:Core",
"@llvm-project//llvm:Support",
],
)
gentbl(
name = "LLVMAVX512ConversionIncGen",
strip_include_prefix = "include",
tbl_outs = [
(
"-gen-llvmir-conversions",
"include/mlir/Dialect/LLVMIR/LLVMAVX512Conversions.inc",
),
],
tblgen = ":mlir-tblgen",
td_file = "include/mlir/Dialect/LLVMIR/LLVMAVX512.td",
td_srcs = [
":LLVMAVX512TdFiles",
],
)
cc_library(
name = "TargetLLVMAVX512Intr",
srcs = [
"lib/Target/LLVMIR/LLVMAVX512Intr.cpp",
],
includes = ["include"],
deps = [
":IR",
":LLVMAVX512",
":LLVMAVX512ConversionIncGen",
":LLVMIRModuleTranslation",
":Translation",
"@llvm-project//llvm:Core",
"@llvm-project//llvm:Support",
],
)
cc_library(
name = "LLVMDialect",
srcs = glob(
[
"lib/Dialect/LLVMIR/IR/*.cpp",
"lib/Dialect/LLVMIR/IR/*.h",
],
exclude = [
"lib/Dialect/LLVMIR/IR/*AVX512*.cpp",
"lib/Dialect/LLVMIR/IR/*AVX512*.h",
"lib/Dialect/LLVMIR/IR/NVVM*.cpp",
"lib/Dialect/LLVMIR/IR/NVVM*.h",
"lib/Dialect/LLVMIR/IR/ROCDL*.cpp",
"lib/Dialect/LLVMIR/IR/ROCDL*.h",
],
),
hdrs = glob(
[
"include/mlir/Dialect/LLVMIR/*.h",
],
exclude = [
"include/mlir/Dialect/LLVMIR/*AVX512*.h",
"include/mlir/Dialect/LLVMIR/NVVM*.h",
"include/mlir/Dialect/LLVMIR/ROCDL*.h",
],
),
includes = ["include"],
deps = [
":ControlFlowInterfaces",
":IR",
":LLVMOpsIncGen",
":SideEffectInterfaces",
":Support",
"@llvm-project//llvm:AsmParser",
"@llvm-project//llvm:BitReader",
"@llvm-project//llvm:BitWriter",
"@llvm-project//llvm:Core",
"@llvm-project//llvm:Support",
],
)
gentbl(
name = "LLVMPassIncGen",
strip_include_prefix = "include",
tbl_outs = [
(
"-gen-pass-decls",
"include/mlir/Dialect/LLVMIR/Transforms/Passes.h.inc",
),
],
tblgen = ":mlir-tblgen",
td_file = "include/mlir/Dialect/LLVMIR/Transforms/Passes.td",
td_srcs = [
":PassBaseTdFiles",
],
)
cc_library(
name = "LLVMIRTransforms",
srcs = glob([
"lib/Dialect/LLVMIR/Transforms/*.cpp",
"lib/Dialect/LLVMIR/Transforms/*.h",
]),
hdrs = glob(["include/mlir/Dialect/LLVMIR/Transforms/*.h"]),
includes = ["include"],
deps = [
":IR",
":LLVMDialect",
":LLVMPassIncGen",
":Pass",
],
)
filegroup(
name = "GPUOpsTdFiles",
srcs = [
"include/mlir/Dialect/GPU/GPUBase.td",
"include/mlir/Dialect/GPU/GPUOps.td",
"include/mlir/Dialect/LLVMIR/LLVMOpBase.td",
"include/mlir/IR/SymbolInterfaces.td",
"include/mlir/Interfaces/SideEffectInterfaces.td",
":OpBaseTdFiles",
],
)
gentbl(
name = "ParallelLoopMapperAttrGen",
strip_include_prefix = "include",
tbl_outs = [
(
"-gen-struct-attr-decls",
"include/mlir/Dialect/GPU/ParallelLoopMapperAttr.h.inc",
),
(
"-gen-struct-attr-defs",
"include/mlir/Dialect/GPU/ParallelLoopMapperAttr.cpp.inc",
),
(
"-gen-enum-decls",
"include/mlir/Dialect/GPU/ParallelLoopMapperEnums.h.inc",
),
(
"-gen-enum-defs",
"include/mlir/Dialect/GPU/ParallelLoopMapperEnums.cpp.inc",
),
],
tblgen = ":mlir-tblgen",
td_file = "include/mlir/Dialect/GPU/ParallelLoopMapperAttr.td",
td_srcs = [
":GPUOpsTdFiles",
":AffineOpsTdFiles",
],
)
gentbl(
name = "GPUOpsIncGen",
strip_include_prefix = "include",
tbl_outs = [
(
"-gen-op-decls",
"include/mlir/Dialect/GPU/GPUOps.h.inc",
),
(
"-gen-op-defs",
"include/mlir/Dialect/GPU/GPUOps.cpp.inc",
),
(
"-gen-dialect-decls -dialect=gpu",
"include/mlir/Dialect/GPU/GPUOpsDialect.h.inc",
),
],
tblgen = ":mlir-tblgen",
td_file = "include/mlir/Dialect/GPU/GPUOps.td",
td_srcs = [
":GPUOpsTdFiles",
],
)
cc_library(
name = "GPUDialect",
srcs = glob(
[
"lib/Dialect/GPU/IR/*.cpp",
"lib/Dialect/GPU/IR/*.h",
],
),
hdrs = glob([
"include/mlir/Dialect/GPU/GPUDialect.h",
]),
includes = ["include"],
deps = [
":GPUOpsIncGen",
":IR",
":LLVMDialect",
":SideEffectInterfaces",
":StandardOps",
":Support",
],
)
gentbl(
name = "GPUPassIncGen",
strip_include_prefix = "include",
tbl_outs = [
(
"-gen-pass-decls",
"include/mlir/Dialect/GPU/Passes.h.inc",
),
],
tblgen = ":mlir-tblgen",
td_file = "include/mlir/Dialect/GPU/Passes.td",
td_srcs = [
":PassBaseTdFiles",
],
)
cc_library(
name = "GPUTransforms",
srcs = glob(
[
"lib/Dialect/GPU/Transforms/*.cpp",
"lib/Dialect/GPU/Transforms/*.h",
],
),
hdrs = [
"include/mlir/Dialect/GPU/MemoryPromotion.h",
"include/mlir/Dialect/GPU/ParallelLoopMapper.h",
"include/mlir/Dialect/GPU/Passes.h",
"include/mlir/Dialect/GPU/Utils.h",
],
includes = ["include"],
deps = [
":EDSC",
":GPUDialect",
":GPUPassIncGen",
":IR",
":ParallelLoopMapperAttrGen",
":Pass",
":SCFDialect",
":StandardOps",
":Support",
":Transforms",
"@llvm-project//llvm:Support",
],
)
filegroup(
name = "LLVMOpsTdFiles",
srcs = [
"include/mlir/Dialect/LLVMIR/LLVMOpBase.td",
"include/mlir/Dialect/LLVMIR/LLVMOps.td",
"include/mlir/IR/SymbolInterfaces.td",
"include/mlir/Interfaces/ControlFlowInterfaces.td",
"include/mlir/Interfaces/SideEffectInterfaces.td",
":OpBaseTdFiles",
],
)
cc_library(
name = "GPUCommonTransforms",
hdrs = [
"lib/Conversion/GPUCommon/IndexIntrinsicsOpLowering.h",
"lib/Conversion/GPUCommon/OpToFuncCallLowering.h",
],
# TODO(b/155492113): Move back to hdrs once fixed.
textual_hdrs = [
"lib/Conversion/GPUCommon/GPUOpsLowering.h",
],
deps = [
":GPUDialect",
":IR",
":LLVMDialect",
":StandardOps",
":StandardToLLVM",
"@llvm-project//llvm:Support",
],
)
gentbl(
name = "GPUToNVVMGen",
strip_include_prefix = "lib/Conversion/GPUToNVVM",
tbl_outs = [
(
"-gen-rewriters",
"lib/Conversion/GPUToNVVM/GPUToNVVM.cpp.inc",
),
],
tblgen = ":mlir-tblgen",
td_file = "lib/Conversion/GPUToNVVM/GPUToNVVM.td",
td_srcs = [
":GPUOpsTdFiles",
":NVVMOpsTdFiles",
],
)
cc_library(
name = "GPUToNVVMTransforms",
srcs = glob([
"lib/Conversion/GPUToNVVM/*.cpp",
"lib/Conversion/GPUToNVVM/*.h",
]) + ["lib/Conversion/PassDetail.h"],
hdrs = glob([
"include/mlir/Conversion/GPUToNVVM/*.h",
]),
includes = ["include"],
deps = [
":ConversionPassIncGen",
":GPUCommonTransforms",
":GPUDialect",
":GPUToNVVMGen",
":GPUTransforms",
":IR",
":NVVMDialect",
":Pass",
":StandardToLLVM",
":Transforms",
"@llvm-project//llvm:Support",
],
)
cc_library(
name = "VectorToROCDL",
srcs = [
"lib/Conversion/PassDetail.h",
"lib/Conversion/VectorToROCDL/VectorToROCDL.cpp",
],
hdrs = [
"include/mlir/Conversion/VectorToROCDL/VectorToROCDL.h",
],
includes = ["include"],
deps = [
":ConversionPassIncGen",
":GPUDialect",
":LLVMDialect",
":Pass",
":ROCDLDialect",
":StandardOps",
":StandardToLLVM",
":Transforms",
":VectorOps",
],
)
gentbl(
name = "GPUToROCDLTGen",
strip_include_prefix = "lib/Conversion/GPUToROCDL",
tbl_outs = [
(
"-gen-rewriters",
"lib/Conversion/GPUToROCDL/GPUToROCDL.cpp.inc",
),
],
tblgen = ":mlir-tblgen",
td_file = "lib/Conversion/GPUToROCDL/GPUToROCDL.td",
td_srcs = [
":GPUOpsTdFiles",
":ROCDLOpsTdFiles",
],
)
cc_library(
name = "GPUToROCDLTransforms",
srcs = [
"lib/Conversion/GPUToROCDL/LowerGpuOpsToROCDLOps.cpp",
"lib/Conversion/PassDetail.h",
],
hdrs = [
"include/mlir/Conversion/GPUToROCDL/GPUToROCDLPass.h",
],
includes = ["include"],
deps = [
":ConversionPassIncGen",
":GPUCommonTransforms",
":GPUDialect",
":GPUToROCDLTGen",
":GPUTransforms",
":Pass",
":ROCDLDialect",
":StandardToLLVM",
":Transforms",
":VectorOps",
":VectorToLLVM",
":VectorToROCDL",
":VectorToSCF",
"@llvm-project//llvm:Support",
],
)
cc_library(
name = "GPUToVulkanTransforms",
srcs = [
"lib/Conversion/GPUToVulkan/ConvertGPULaunchFuncToVulkanLaunchFunc.cpp",
"lib/Conversion/GPUToVulkan/ConvertLaunchFuncToVulkanCalls.cpp",
"lib/Conversion/PassDetail.h",
],
hdrs = ["include/mlir/Conversion/GPUToVulkan/ConvertGPUToVulkanPass.h"],
includes = ["include"],
deps = [
":ConversionPassIncGen",
":GPUDialect",
":IR",
":LLVMDialect",
":Pass",
":SPIRVDialect",
":SPIRVSerialization",
":StandardOps",
":Support",
"@llvm-project//llvm:Support",
],
)
cc_library(
name = "GPUToGPURuntimeTransforms",
srcs = [
"lib/Conversion/GPUCommon/ConvertKernelFuncToBlob.cpp",
"lib/Conversion/GPUCommon/ConvertLaunchFuncToRuntimeCalls.cpp",
"lib/Conversion/PassDetail.h",
],
hdrs = ["include/mlir/Conversion/GPUCommon/GPUCommonPass.h"],
includes = ["include"],
deps = [
":ConversionPassIncGen",
":GPUDialect",
":IR",
":LLVMDialect",
":Pass",
":Support",
":TargetNVVMIR",
"@llvm-project//llvm:Core",
"@llvm-project//llvm:NVPTXCodeGen",
"@llvm-project//llvm:Support",
"@llvm-project//llvm:Target",
],
)
gentbl(
name = "GPUToSPIRVIncGen",
strip_include_prefix = "lib/Conversion/GPUToSPIRV",
tbl_outs = [
(
"-gen-rewriters",
"lib/Conversion/GPUToSPIRV/GPUToSPIRV.cpp.inc",
),
],
tblgen = ":mlir-tblgen",
td_file = "lib/Conversion/GPUToSPIRV/GPUToSPIRV.td",
td_srcs = [
":GPUOpsTdFiles",
":SPIRVOpsTdFiles",
],
)
cc_library(
name = "GPUToSPIRVTransforms",
srcs = [
"lib/Conversion/GPUToSPIRV/ConvertGPUToSPIRV.cpp",
"lib/Conversion/GPUToSPIRV/ConvertGPUToSPIRVPass.cpp",
"lib/Conversion/PassDetail.h",
],
hdrs = [
"include/mlir/Conversion/GPUToSPIRV/ConvertGPUToSPIRV.h",
"include/mlir/Conversion/GPUToSPIRV/ConvertGPUToSPIRVPass.h",
],
includes = [
"include",
"lib/Conversions/GPUToSPIRV",
],
deps = [
":ConversionPassIncGen",
":GPUDialect",
":GPUToSPIRVIncGen",
":IR",
":Pass",
":SCFDialect",
":SPIRVDialect",
":SPIRVLowering",
":StandardToSPIRVTransforms",
":Support",
":Transforms",
],
)
cc_library(
name = "SPIRVToLLVM",
srcs = glob([
"lib/Conversion/SPIRVToLLVM/*.cpp",
]) + [
"lib/Conversion/PassDetail.h",
],
hdrs = glob([
"include/mlir/Conversion/SPIRVToLLVM/*.h",
]),
includes = ["include"],
deps = [
":ConversionPassIncGen",
":IR",
":LLVMDialect",
":Pass",
":SPIRVDialect",
":StandardOps",
":StandardToLLVM",
":Support",
":Transforms",
],
)
gentbl(
name = "LLVMOpsIncGen",
strip_include_prefix = "include",
tbl_outs = [
(
"-gen-op-decls",
"include/mlir/Dialect/LLVMIR/LLVMOps.h.inc",
),
(
"-gen-op-defs",
"include/mlir/Dialect/LLVMIR/LLVMOps.cpp.inc",
),
(
"-gen-dialect-decls",
"include/mlir/Dialect/LLVMIR/LLVMOpsDialect.h.inc",
),
(
"-gen-enum-decls",
"include/mlir/Dialect/LLVMIR/LLVMOpsEnums.h.inc",
),
(
"-gen-enum-defs",
"include/mlir/Dialect/LLVMIR/LLVMOpsEnums.cpp.inc",
),
],
tblgen = ":mlir-tblgen",
td_file = "include/mlir/Dialect/LLVMIR/LLVMOps.td",
td_srcs = [
":LLVMOpsTdFiles",
],
)
gentbl(
name = "LLVMConversionIncGen",
strip_include_prefix = "include",
tbl_outs = [
(
"-gen-llvmir-conversions",
"include/mlir/Dialect/LLVMIR/LLVMConversions.inc",
),
(
"-gen-enum-to-llvmir-conversions",
"include/mlir/Dialect/LLVMIR/LLVMConversionEnumsToLLVM.inc",
),
(
"-gen-enum-from-llvmir-conversions",
"include/mlir/Dialect/LLVMIR/LLVMConversionEnumsFromLLVM.inc",
),
],
tblgen = ":mlir-tblgen",
td_file = "include/mlir/Dialect/LLVMIR/LLVMOps.td",
td_srcs = [
":LLVMOpsTdFiles",
],
)
cc_library(
name = "NVVMDialect",
srcs = [
"lib/Dialect/LLVMIR/IR/NVVMDialect.cpp",
],
hdrs = [
"include/mlir/Dialect/LLVMIR/NVVMDialect.h",
],
includes = ["include"],
deps = [
":IR",
":LLVMDialect",
":NVVMOpsIncGen",
":SideEffectInterfaces",
":StandardOps",
":Support",
"@llvm-project//llvm:AsmParser",
"@llvm-project//llvm:Core",
"@llvm-project//llvm:Support",
],
)
filegroup(
name = "NVVMOpsTdFiles",
srcs = [
"include/mlir/Dialect/LLVMIR/LLVMOpBase.td",
"include/mlir/Dialect/LLVMIR/NVVMOps.td",
"include/mlir/Interfaces/SideEffectInterfaces.td",
":OpBaseTdFiles",
],
)
gentbl(
name = "NVVMOpsIncGen",
strip_include_prefix = "include",
tbl_outs = [
(
"-gen-op-decls",
"include/mlir/Dialect/LLVMIR/NVVMOps.h.inc",
),
(
"-gen-op-defs",
"include/mlir/Dialect/LLVMIR/NVVMOps.cpp.inc",
),
(
"-gen-dialect-decls -dialect=nvvm",
"include/mlir/Dialect/LLVMIR/NVVMOpsDialect.h.inc",
),
],
tblgen = ":mlir-tblgen",
td_file = "include/mlir/Dialect/LLVMIR/NVVMOps.td",
td_srcs = [
":NVVMOpsTdFiles",
],
)
gentbl(
name = "NVVMConversionIncGen",
strip_include_prefix = "include",
tbl_outs = [
(
"-gen-llvmir-conversions",
"include/mlir/Dialect/LLVMIR/NVVMConversions.inc",
),
],
tblgen = ":mlir-tblgen",
td_file = "include/mlir/Dialect/LLVMIR/NVVMOps.td",
td_srcs = [
":NVVMOpsTdFiles",
],
)
cc_library(
name = "ROCDLDialect",
srcs = [
"lib/Dialect/LLVMIR/IR/ROCDLDialect.cpp",
],
hdrs = [
"include/mlir/Dialect/LLVMIR/ROCDLDialect.h",
],
includes = ["include"],
deps = [
":IR",
":LLVMDialect",
":ROCDLOpsIncGen",
":SideEffectInterfaces",
":StandardOps",
":Support",
"@llvm-project//llvm:AsmParser",
"@llvm-project//llvm:Core",
"@llvm-project//llvm:Support",
],
)
filegroup(
name = "ROCDLOpsTdFiles",
srcs = [
"include/mlir/Dialect/LLVMIR/LLVMOpBase.td",
"include/mlir/Dialect/LLVMIR/ROCDLOps.td",
"include/mlir/Interfaces/SideEffectInterfaces.td",
":OpBaseTdFiles",
],
)
gentbl(
name = "ROCDLOpsIncGen",
strip_include_prefix = "include",
tbl_outs = [
(
"-gen-op-decls",
"include/mlir/Dialect/LLVMIR/ROCDLOps.h.inc",
),
(
"-gen-op-defs",
"include/mlir/Dialect/LLVMIR/ROCDLOps.cpp.inc",
),
(
"-gen-dialect-decls -dialect=rocdl",
"include/mlir/Dialect/LLVMIR/ROCDLOpsDialect.h.inc",
),
],
tblgen = ":mlir-tblgen",
td_file = "include/mlir/Dialect/LLVMIR/ROCDLOps.td",
td_srcs = [
":ROCDLOpsTdFiles",
],
)
gentbl(
name = "ROCDLConversionIncGen",
strip_include_prefix = "include",
tbl_outs = [
(
"-gen-llvmir-conversions",
"include/mlir/Dialect/LLVMIR/ROCDLConversions.inc",
),
],
tblgen = ":mlir-tblgen",
td_file = "include/mlir/Dialect/LLVMIR/ROCDLOps.td",
td_srcs = [
":ROCDLOpsTdFiles",
],
)
# TODO(gcmn): Update SPIRV dependencies so that they map better to cmake files.
filegroup(
name = "SPIRVOpsTdFiles",
srcs = [
"include/mlir/IR/SymbolInterfaces.td",
"include/mlir/Interfaces/CallInterfaces.td",
"include/mlir/Interfaces/ControlFlowInterfaces.td",
"include/mlir/Interfaces/SideEffectInterfaces.td",
":OpBaseTdFiles",
] + glob(["include/mlir/Dialect/SPIRV/*.td"]),
)
gentbl(
name = "SPIRVOpsIncGen",
strip_include_prefix = "include",
tbl_outs = [
(
"-gen-op-decls",
"include/mlir/Dialect/SPIRV/SPIRVOps.h.inc",
),
(
"-gen-op-defs",
"include/mlir/Dialect/SPIRV/SPIRVOps.cpp.inc",
),
(
"-gen-dialect-decls",
"include/mlir/Dialect/SPIRV/SPIRVOpsDialect.h.inc",
),
(
"-gen-op-doc",
"g3doc/Dialects/SPIRV/SPIRVOps.md",
),
(
"-gen-enum-decls",
"include/mlir/Dialect/SPIRV/SPIRVEnums.h.inc",
),
(
"-gen-enum-defs",
"include/mlir/Dialect/SPIRV/SPIRVEnums.cpp.inc",
),
(
"-gen-spirv-enum-avail-decls",
"include/mlir/Dialect/SPIRV/SPIRVEnumAvailability.h.inc",
),
(
"-gen-spirv-enum-avail-defs",
"include/mlir/Dialect/SPIRV/SPIRVEnumAvailability.cpp.inc",
),
(
"-gen-spirv-capability-implication",
"include/mlir/Dialect/SPIRV/SPIRVCapabilityImplication.inc",
),
],
tblgen = ":mlir-tblgen",
td_file = "include/mlir/Dialect/SPIRV/SPIRVOps.td",
td_srcs = [
":SPIRVOpsTdFiles",
],
)
gentbl(
name = "SPIRVCanonicalizationIncGen",
strip_include_prefix = "lib/Dialect/SPIRV",
tbl_outs = [
(
"-gen-rewriters",
"lib/Dialect/SPIRV/SPIRVCanonicalization.inc",
),
],
tblgen = ":mlir-tblgen",
td_file = "lib/Dialect/SPIRV/SPIRVCanonicalization.td",
td_srcs = [
":SPIRVOpsTdFiles",
"lib/Dialect/SPIRV/SPIRVCanonicalization.td",
],
)
gentbl(
name = "SPIRVAvailabilityIncGen",
strip_include_prefix = "include",
tbl_outs = [
(
"-gen-avail-interface-decls",
"include/mlir/Dialect/SPIRV/SPIRVAvailability.h.inc",
),
(
"-gen-avail-interface-defs",
"include/mlir/Dialect/SPIRV/SPIRVAvailability.cpp.inc",
),
(
"-gen-spirv-avail-impls",
"include/mlir/Dialect/SPIRV/SPIRVOpAvailabilityImpl.inc",
),
],
tblgen = ":mlir-tblgen",
td_file = "include/mlir/Dialect/SPIRV/SPIRVOps.td",
td_srcs = [
":SPIRVOpsTdFiles",
],
)
gentbl(
name = "SPIRVTargetAndABIStructGen",
tbl_outs = [
(
"-gen-struct-attr-decls",
"include/mlir/Dialect/SPIRV/TargetAndABI.h.inc",
),
(
"-gen-struct-attr-defs",
"include/mlir/Dialect/SPIRV/TargetAndABI.cpp.inc",
),
],
tblgen = ":mlir-tblgen",
td_file = "include/mlir/Dialect/SPIRV/TargetAndABI.td",
td_srcs = [
":SPIRVOpsTdFiles",
":StdOpsTdFiles",
],
)
gentbl(
name = "SPIRVOpUtilsIncGen",
strip_include_prefix = "include",
tbl_outs = [
(
"-gen-spirv-op-utils",
"include/mlir/Dialect/SPIRV/SPIRVOpUtils.inc",
),
],
tblgen = ":mlir-tblgen",
td_file = "include/mlir/Dialect/SPIRV/SPIRVBase.td",
td_srcs = [
":SPIRVOpsTdFiles",
":SPIRVAvailabilityIncGen",
],
)
gentbl(
name = "SPIRVSerializationGen",
strip_include_prefix = "include",
tbl_outs = [
(
"-gen-spirv-serialization",
"include/mlir/Dialect/SPIRV/SPIRVSerialization.inc",
),
],
tblgen = ":mlir-tblgen",
td_file = "include/mlir/Dialect/SPIRV/SPIRVOps.td",
td_srcs = [
":SPIRVOpsTdFiles",
],
)
cc_library(
name = "SPIRVDialect",
srcs = glob(
[
"lib/Dialect/SPIRV/*.cpp",
"lib/Dialect/SPIRV/*.h",
],
exclude = [
"lib/Dialect/SPIRV/SPIRVLowering.cpp",
],
) + [
"include/mlir/Transforms/InliningUtils.h",
],
hdrs = glob(
[
"include/mlir/Dialect/SPIRV/*.h",
],
exclude = [
"include/mlir/Dialect/SPIRV/SPIRVBinaryUtils.h",
"include/mlir/Dialect/SPIRV/SPIRVLowering.h",
],
),
includes = ["include"],
deps = [
":CommonFolders",
":ControlFlowInterfaces",
":IR",
":Parser",
":Pass",
":SPIRVAvailabilityIncGen",
":SPIRVCanonicalizationIncGen",
":SPIRVOpUtilsIncGen",
":SPIRVOpsIncGen",
":SPIRVSerializationGen",
":SPIRVTargetAndABIStructGen",
":SideEffectInterfaces",
":Support",
":Transforms",
"@llvm-project//llvm:Support",
],
)
gentbl(
name = "SPIRVPassIncGen",
strip_include_prefix = "include",
tbl_outs = [
(
"-gen-pass-decls",
"include/mlir/Dialect/SPIRV/Passes.h.inc",
),
],
tblgen = ":mlir-tblgen",
td_file = "include/mlir/Dialect/SPIRV/Passes.td",
td_srcs = [
":PassBaseTdFiles",
],
)
cc_library(
name = "SPIRVLowering",
srcs = glob([
"lib/Dialect/SPIRV/Transforms/*.cpp",
"lib/Dialect/SPIRV/Transforms/*.h",
]) + [
"lib/Dialect/SPIRV/SPIRVLowering.cpp",
],
hdrs = [
"include/mlir/Dialect/SPIRV/Passes.h",
"include/mlir/Dialect/SPIRV/SPIRVLowering.h",
"include/mlir/Dialect/SPIRV/TargetAndABI.h",
],
includes = [
"include",
],
deps = [
":IR",
":Pass",
":SPIRVDialect",
":SPIRVPassIncGen",
":SPIRVTargetAndABIStructGen",
":StandardOps",
":Support",
":Transforms",
"@llvm-project//llvm:Support",
],
)
cc_library(
name = "StandardToSPIRVTransforms",
srcs = glob([
"lib/Conversion/StandardToSPIRV/*.cpp",
"lib/Conversion/StandardToSPIRV/*.h",
]) + ["lib/Conversion/PassDetail.h"],
hdrs = glob([
"include/mlir/Conversion/StandardToSPIRV/*.h",
]),
includes = [
"include",
"lib/Conversion/StandardToSPIRV",
],
deps = [
":ConversionPassIncGen",
":IR",
":Pass",
":SPIRVDialect",
":SPIRVLowering",
":StandardOps",
":Support",
":Transforms",
"@llvm-project//llvm:Support",
],
)
alias(
name = "StandardToSPIRVConversions",
actual = "StandardToSPIRVTransforms",
)
cc_library(
name = "SPIRVSerialization",
srcs = glob(
[
"lib/Dialect/SPIRV/Serialization/*.cpp",
],
exclude = [
"lib/Dialect/SPIRV/Serialization/TranslateRegistration.cpp",
],
),
hdrs = [
"include/mlir/Dialect/SPIRV/SPIRVBinaryUtils.h",
"include/mlir/Dialect/SPIRV/Serialization.h",
],
includes = ["include"],
deps = [
":IR",
":SPIRVDialect",
":SPIRVOpUtilsIncGen",
":SPIRVOpsIncGen",
":SPIRVSerializationGen",
":Support",
":Transforms",
"@llvm-project//llvm:Support",
],
)
cc_library(
name = "SPIRVTranslateRegistration",
srcs = [
"lib/Dialect/SPIRV/Serialization/TranslateRegistration.cpp",
],
includes = ["include"],
deps = [
":IR",
":Parser",
":SPIRVDialect",
":SPIRVSerialization",
":Support",
":Translation",
"@llvm-project//llvm:Support",
],
)
cc_library(
name = "TransformUtils",
srcs = glob([
"lib/Transforms/Utils/*.cpp",
"lib/Transforms/Utils/*.h",
]),
hdrs = glob([
"include/mlir/Transforms/*.h",
]),
includes = ["include"],
deps = [
":Affine",
":Analysis",
":ControlFlowInterfaces",
":IR",
":LoopLikeInterface",
":SCFDialect",
":SideEffectInterfaces",
":StandardOps",
":Support",
"@llvm-project//llvm:Support",
],
)
gentbl(
name = "DerivedAttributeOpInterfaceIncGen",
strip_include_prefix = "include",
tbl_outs = [
(
"-gen-op-interface-decls",
"include/mlir/Interfaces/DerivedAttributeOpInterface.h.inc",
),
(
"-gen-op-interface-defs",
"include/mlir/Interfaces/DerivedAttributeOpInterface.cpp.inc",
),
],
tblgen = ":mlir-tblgen",
td_file = "include/mlir/Interfaces/DerivedAttributeOpInterface.td",
td_srcs = [
":OpBaseTdFiles",
],
)
cc_library(
name = "DerivedAttributeOpInterface",
srcs = [
"lib/Interfaces/DerivedAttributeOpInterface.cpp",
],
hdrs = [
"include/mlir/Interfaces/DerivedAttributeOpInterface.h",
],
includes = ["include"],
deps = [
":DerivedAttributeOpInterfaceIncGen",
":IR",
":Support",
"@llvm-project//llvm:Support",
],
)
gentbl(
name = "LoopLikeInterfaceIncGen",
strip_include_prefix = "include",
tbl_outs = [
(
"-gen-op-interface-decls",
"include/mlir/Interfaces/LoopLikeInterface.h.inc",
),
(
"-gen-op-interface-defs",
"include/mlir/Interfaces/LoopLikeInterface.cpp.inc",
),
],
tblgen = ":mlir-tblgen",
td_file = "include/mlir/Interfaces/LoopLikeInterface.td",
td_srcs = [
":OpBaseTdFiles",
],
)
gentbl(
name = "ViewLikeInterfaceIncGen",
strip_include_prefix = "include",
tbl_outs = [
(
"-gen-op-interface-decls",
"include/mlir/Interfaces/ViewLikeInterface.h.inc",
),
(
"-gen-op-interface-defs",
"include/mlir/Interfaces/ViewLikeInterface.cpp.inc",
),
],
tblgen = ":mlir-tblgen",
td_file = "include/mlir/Interfaces/ViewLikeInterface.td",
td_srcs = [
":OpBaseTdFiles",
],
)
gentbl(
name = "TransformsPassIncGen",
strip_include_prefix = "include",
tbl_outs = [
(
"-gen-pass-decls",
"include/mlir/Transforms/Passes.h.inc",
),
],
tblgen = ":mlir-tblgen",
td_file = "include/mlir/Transforms/Passes.td",
td_srcs = [
":PassBaseTdFiles",
],
)
cc_library(
name = "Transforms",
srcs = glob([
"lib/Transforms/*.cpp",
"lib/Transforms/*.h",
]),
hdrs = glob([
"include/mlir/Transforms/*.h",
]),
includes = ["include"],
deps = [
":Affine",
":Analysis",
":ControlFlowInterfaces",
":IR",
":LinalgOps",
":LoopLikeInterface",
":Pass",
":SCFDialect",
":SideEffectInterfaces",
":StandardOps",
":Support",
":TransformUtils",
":TransformsPassIncGen",
"@llvm-project//llvm:Support",
],
)
cc_library(
name = "CommonFolders",
srcs = [
],
hdrs = [
"include/mlir/Dialect/CommonFolders.h",
],
includes = ["include"],
deps = [
":IR",
"@llvm-project//llvm:Support",
],
)
cc_library(
name = "SCFToGPU",
srcs = ["lib/Conversion/SCFToGPU/SCFToGPU.cpp"],
hdrs = ["include/mlir/Conversion/SCFToGPU/SCFToGPU.h"],
includes = ["include"],
deps = [
":Affine",
":AffineToStandard",
":ConversionPassIncGen",
":GPUDialect",
":GPUTransforms",
":IR",
":Pass",
":SCFDialect",
":StandardOps",
":Support",
":TransformUtils",
":Transforms",
"@llvm-project//llvm:Support",
],
)
cc_library(
name = "SCFToGPUPass",
srcs = [
"lib/Conversion/PassDetail.h",
"lib/Conversion/SCFToGPU/SCFToGPUPass.cpp",
],
hdrs = [
"include/mlir/Conversion/SCFToGPU/SCFToGPUPass.h",
],
includes = ["include"],
deps = [
":Affine",
":ConversionPassIncGen",
":GPUDialect",
":Pass",
":SCFDialect",
":SCFToGPU",
":StandardOps",
":Support",
":Transforms",
"@llvm-project//llvm:Support",
],
)
cc_library(
name = "SCFToStandard",
srcs = [
"lib/Conversion/PassDetail.h",
"lib/Conversion/SCFToStandard/SCFToStandard.cpp",
],
hdrs = [
"include/mlir/Conversion/SCFToStandard/SCFToStandard.h",
],
includes = ["include"],
deps = [
":ConversionPassIncGen",
":IR",
":LLVMDialect",
":Pass",
":SCFDialect",
":StandardOps",
":Support",
":TransformUtils",
":Transforms",
],
)
alias(
name = "CFGTransforms",
actual = "SCFToStandard",
)
cc_library(
name = "StandardToLLVM",
srcs = [
"lib/Conversion/PassDetail.h",
"lib/Conversion/StandardToLLVM/StandardToLLVM.cpp",
],
hdrs = [
"include/mlir/Conversion/StandardToLLVM/ConvertStandardToLLVM.h",
"include/mlir/Conversion/StandardToLLVM/ConvertStandardToLLVMPass.h",
],
includes = ["include"],
deps = [
":ConversionPassIncGen",
":IR",
":LLVMDialect",
":Pass",
":StandardOps",
":Support",
":TransformUtils",
":Transforms",
"@llvm-project//llvm:Core",
"@llvm-project//llvm:Support",
],
)
alias(
name = "LLVMTransforms",
actual = "StandardToLLVM",
)
gentbl(
name = "CallOpInterfacesIncGen",
strip_include_prefix = "include",
tbl_outs = [
(
"-gen-op-interface-decls",
"include/mlir/Interfaces/CallInterfaces.h.inc",
),
(
"-gen-op-interface-defs",
"include/mlir/Interfaces/CallInterfaces.cpp.inc",
),
],
tblgen = ":mlir-tblgen",
td_file = "include/mlir/Interfaces/CallInterfaces.td",
td_srcs = [
":OpBaseTdFiles",
],
)
cc_library(
name = "CallOpInterfaces",
srcs = [
"lib/Interfaces/CallInterfaces.cpp",
],
hdrs = [
"include/mlir/Interfaces/CallInterfaces.h",
],
includes = ["include"],
deps = [
":CallOpInterfacesIncGen",
":IR",
":Support",
"@llvm-project//llvm:Support",
],
)
gentbl(
name = "ControlFlowInterfacesIncGen",
strip_include_prefix = "include",
tbl_outs = [
(
"-gen-op-interface-decls",
"include/mlir/Interfaces/ControlFlowInterfaces.h.inc",
),
(
"-gen-op-interface-defs",
"include/mlir/Interfaces/ControlFlowInterfaces.cpp.inc",
),
],
tblgen = ":mlir-tblgen",
td_file = "include/mlir/Interfaces/ControlFlowInterfaces.td",
td_srcs = [
":OpBaseTdFiles",
],
)
cc_library(
name = "ControlFlowInterfaces",
srcs = [
"lib/Interfaces/ControlFlowInterfaces.cpp",
],
hdrs = [
"include/mlir/Interfaces/ControlFlowInterfaces.h",
],
includes = ["include"],
deps = [
":ControlFlowInterfacesIncGen",
":IR",
":Support",
"@llvm-project//llvm:Support",
],
)
gentbl(
name = "InferTypeOpInterfaceIncGen",
strip_include_prefix = "include",
tbl_outs = [
(
"-gen-op-interface-decls",
"include/mlir/Interfaces/InferTypeOpInterface.h.inc",
),
(
"-gen-op-interface-defs",
"include/mlir/Interfaces/InferTypeOpInterface.cpp.inc",
),
],
tblgen = ":mlir-tblgen",
td_file = "include/mlir/Interfaces/InferTypeOpInterface.td",
td_srcs = [
":OpBaseTdFiles",
],
)
cc_library(
name = "InferTypeOpInterface",
srcs = [
"lib/Interfaces/InferTypeOpInterface.cpp",
],
hdrs = [
"include/mlir/Interfaces/InferTypeOpInterface.h",
],
includes = ["include"],
deps = [
":IR",
":InferTypeOpInterfaceIncGen",
":Support",
"@llvm-project//llvm:Support",
],
)
gentbl(
name = "SideEffectInterfacesIncGen",
strip_include_prefix = "include",
tbl_outs = [
(
"-gen-op-interface-decls",
"include/mlir/Interfaces/SideEffectInterfaces.h.inc",
),
(
"-gen-op-interface-defs",
"include/mlir/Interfaces/SideEffectInterfaces.cpp.inc",
),
],
tblgen = ":mlir-tblgen",
td_file = "include/mlir/Interfaces/SideEffectInterfaces.td",
td_srcs = [
":OpBaseTdFiles",
],
)
cc_library(
name = "SideEffectInterfaces",
srcs = [
"lib/Interfaces/SideEffectInterfaces.cpp",
],
hdrs = [
"include/mlir/Interfaces/SideEffectInterfaces.h",
],
includes = ["include"],
deps = [
":IR",
":SideEffectInterfacesIncGen",
":Support",
"@llvm-project//llvm:Support",
],
)
alias(
name = "SideEffects",
actual = "SideEffectInterfaces",
)
cc_library(
name = "Analysis",
srcs = glob(
[
"lib/Analysis/*.cpp",
"lib/Analysis/*.h",
],
exclude = [
"lib/Analysis/Vector*.cpp",
"lib/Analysis/Vector*.h",
],
),
hdrs = glob(
[
"include/mlir/Analysis/*.h",
],
exclude = [
"include/mlir/Analysis/Vector*.h",
],
),
includes = ["include"],
deps = [
":Affine",
":CallOpInterfaces",
":IR",
":SCFDialect",
":StandardOps",
":Support",
"@llvm-project//llvm:Support",
],
)
cc_library(
name = "Translation",
srcs = glob([
"lib/Translation/*.cpp",
"lib/Translation/*.h",
]),
hdrs = [
"include/mlir/Translation.h",
],
includes = ["include"],
deps = [
":Analysis",
":IR",
":Parser",
":Support",
"@llvm-project//llvm:Support",
],
)
cc_library(
name = "LLVMIRModuleTranslation",
srcs = [
"lib/Target/LLVMIR/DebugTranslation.cpp",
"lib/Target/LLVMIR/DebugTranslation.h",
"lib/Target/LLVMIR/ModuleTranslation.cpp",
],
hdrs = [
"include/mlir/Target/LLVMIR/ModuleTranslation.h",
],
includes = ["include"],
deps = [
":IR",
":LLVMConversionIncGen",
":LLVMDialect",
":LLVMIRTransforms",
":OpenMPDialect",
":Support",
"@llvm-project//llvm:Core",
"@llvm-project//llvm:FrontendOpenMP",
"@llvm-project//llvm:Support",
"@llvm-project//llvm:TransformUtils",
],
)
cc_library(
name = "TargetLLVMIR",
srcs = [
"lib/Target/LLVMIR/ConvertFromLLVMIR.cpp",
"lib/Target/LLVMIR/ConvertToLLVMIR.cpp",
],
hdrs = [
"include/mlir/Target/LLVMIR.h",
],
includes = ["include"],
deps = [
":IR",
":LLVMConversionIncGen",
":LLVMDialect",
":LLVMIRModuleTranslation",
":Support",
":TargetLLVMAVX512Intr",
":Translation",
"@llvm-project//llvm:Core",
"@llvm-project//llvm:IRReader",
"@llvm-project//llvm:Support",
],
)
cc_library(
name = "TargetNVVMIR",
srcs = [
"lib/Target/LLVMIR/ConvertToNVVMIR.cpp",
],
hdrs = [
"include/mlir/Target/NVVMIR.h",
],
includes = ["include"],
deps = [
":GPUDialect",
":IR",
":LLVMDialect",
":LLVMIRModuleTranslation",
":NVVMConversionIncGen",
":NVVMDialect",
":Support",
":Translation",
"@llvm-project//llvm:Core",
"@llvm-project//llvm:Support",
],
)
cc_library(
name = "TargetROCDLIR",
srcs = [
"lib/Target/LLVMIR/ConvertToROCDLIR.cpp",
],
hdrs = [
"include/mlir/Target/ROCDLIR.h",
],
includes = ["include"],
deps = [
":GPUDialect",
":IR",
":LLVMDialect",
":LLVMIRModuleTranslation",
":ROCDLConversionIncGen",
":ROCDLDialect",
":Support",
":Translation",
"@llvm-project//llvm:Core",
"@llvm-project//llvm:Support",
],
)
# TODO(zinenko): Update these so that we can simplify mapping to cmake.
cc_library(
name = "ExecutionEngine",
srcs = [
"lib/ExecutionEngine/ExecutionEngine.cpp",
],
hdrs = [
"include/mlir/ExecutionEngine/ExecutionEngine.h",
],
includes = ["include"],
deps = [
":IR",
":LLVMDialect",
":Support",
":TargetLLVMIR",
":Translation",
"@llvm-project//llvm:BitReader",
"@llvm-project//llvm:BitWriter",
"@llvm-project//llvm:Core",
"@llvm-project//llvm:ExecutionEngine",
"@llvm-project//llvm:MC",
"@llvm-project//llvm:OrcJIT",
"@llvm-project//llvm:Support",
"@llvm-project//llvm:Target", # fixdeps: keep
"@llvm-project//llvm:TransformUtils",
"@llvm-project//llvm:X86CodeGen", # fixdeps: keep
"@llvm-project//llvm:X86Disassembler", # fixdeps: keep
],
)
cc_library(
name = "ExecutionEngineUtils",
srcs = [
"lib/ExecutionEngine/OptUtils.cpp",
],
hdrs = [
"include/mlir/ExecutionEngine/OptUtils.h",
],
includes = ["include"],
deps = [
"@llvm-project//llvm:Analysis",
"@llvm-project//llvm:Core",
"@llvm-project//llvm:IPO",
"@llvm-project//llvm:Support",
"@llvm-project//llvm:Target",
],
)
# TODO(jpienaar): Update this.
cc_library(
name = "MlirOptLib",
srcs = [
"lib/Support/MlirOptMain.cpp",
],
includes = ["include"],
deps = [
":Analysis",
":GPUToGPURuntimeTransforms",
":GPUToNVVMTransforms",
":GPUToROCDLTransforms",
":GPUToSPIRVTransforms",
":GPUTransforms",
":IR",
":LLVMDialect",
":LinalgToLLVM",
":LinalgToSPIRV",
":LinalgToStandard",
":NVVMDialect",
":Parser",
":Pass",
":SCFTransforms",
":ShapeToSCF",
":ShapeToStandard",
":ShapeTransforms",
":StandardOpsTransforms",
":StandardToLLVM",
":StandardToSPIRVTransforms",
":Support",
":Transforms",
":VectorToLLVM",
":VectorToSCF",
"@llvm-project//llvm:Support",
"@llvm-project//mlir/test:TestAffine",
"@llvm-project//mlir/test:TestDialect",
"@llvm-project//mlir/test:TestIR",
"@llvm-project//mlir/test:TestPass",
"@llvm-project//mlir/test:TestSPIRV",
"@llvm-project//mlir/test:TestTransforms",
],
)
cc_library(
name = "AllTranslations",
hdrs = ["include/mlir/InitAllTranslations.h"],
deps = [
":SPIRVTranslateRegistration",
":TargetLLVMIR",
":TargetNVVMIR",
":TargetROCDLIR",
],
)
cc_library(
name = "MlirTranslateMain",
srcs = ["tools/mlir-translate/mlir-translate.cpp"],
deps = [
":AllPassesAndDialectsNoRegistration",
":AllTranslations",
":IR",
":Parser",
":Support",
":Translation",
"@llvm-project//llvm:Support",
],
)
cc_binary(
name = "mlir-translate",
deps = [
":MlirTranslateMain",
],
)
cc_library(
name = "AllPassesAndDialectsNoRegistration",
hdrs = [
"include/mlir/InitAllDialects.h",
"include/mlir/InitAllPasses.h",
],
defines = ["MLIR_CUDA_CONVERSIONS_ENABLED"],
deps = [
":AVX512",
":AVX512ToLLVM",
":Affine",
":AffinePassIncGen",
":AffineTransforms",
":ConversionPassIncGen",
":GPUDialect",
":GPUPassIncGen",
":GPUToGPURuntimeTransforms",
":GPUToNVVMTransforms",
":GPUToROCDLTransforms",
":GPUToSPIRVTransforms",
":GPUToVulkanTransforms",
":GPUTransforms",
":IR",
":LLVMAVX512",
":LLVMDialect",
":LLVMIRTransforms",
":LLVMPassIncGen",
":LinalgOps",
":LinalgPassIncGen",
":LinalgToLLVM",
":LinalgToSPIRV",
":LinalgToStandard",
":LinalgTransforms",
":LoopPassIncGen",
":NVVMDialect",
":OpenMPDialect",
":QuantOps",
":QuantPassIncGen",
":ROCDLDialect",
":SCFDialect",
":SCFToGPUPass",
":SCFToStandard",
":SCFTransforms",
":SDBM",
":SPIRVDialect",
":SPIRVLowering",
":SPIRVPassIncGen",
":SPIRVToLLVM",
":Shape",
":ShapeToSCF",
":ShapeToStandard",
":ShapeTransforms",
":ShapeTransformsPassIncGen",
":StandardOps",
":StandardOpsTransforms",
":StandardOpsTransformsPassIncGen",
":StandardToLLVM",
":StandardToSPIRVTransforms",
":Transforms",
":TransformsPassIncGen",
":VectorOps",
":VectorToLLVM",
":VectorToROCDL",
":VectorToSCF",
],
)
cc_library(
name = "AllPassesAndDialects",
srcs = ["@org_tensorflow//third_party/mlir:mlir-auto-init.cpp"],
deps = [
":AllPassesAndDialectsNoRegistration",
],
alwayslink = 1,
)
# TODO(jpienaar): This library should be removed.
cc_library(
name = "MlirOptMain",
srcs = [
"tools/mlir-opt/mlir-opt.cpp",
],
copts = ["-DMLIR_INCLUDE_TESTS"],
deps = [
":AllPassesAndDialectsNoRegistration",
":Analysis",
":IR",
":MlirOptLib",
":Pass",
":Support",
"@llvm-project//llvm:Support",
],
)
cc_binary(
name = "mlir-opt",
deps = [
":Analysis",
":IR",
":MlirOptLib",
":MlirOptMain",
":OpenMPDialect",
":QuantOps",
":SCFToGPUPass",
":Transforms",
"@llvm-project//llvm:AllTargetsCodeGens",
"@llvm-project//llvm:Support",
"@llvm-project//mlir/test:TestAffine",
"@llvm-project//mlir/test:TestDialect",
"@llvm-project//mlir/test:TestIR",
"@llvm-project//mlir/test:TestPass",
"@llvm-project//mlir/test:TestSPIRV",
"@llvm-project//mlir/test:TestTransforms",
],
)
cc_library(
name = "MlirJitRunner",
srcs = ["lib/ExecutionEngine/JitRunner.cpp"],
hdrs = ["include/mlir/ExecutionEngine/JitRunner.h"],
includes = ["include"],
deps = [
":AllPassesAndDialectsNoRegistration",
":ExecutionEngine",
":ExecutionEngineUtils",
":IR",
":LLVMDialect",
":Parser",
":Pass",
":SCFToStandard",
":Support",
"@llvm-project//llvm:Core",
"@llvm-project//llvm:OrcJIT",
"@llvm-project//llvm:Support",
],
)
cc_library(
name = "mlir_c_runner_utils",
srcs = [
"lib/ExecutionEngine/CRunnerUtils.cpp",
],
hdrs = [
"include/mlir/ExecutionEngine/CRunnerUtils.h",
],
includes = ["include"],
)
cc_library(
name = "mlir_runner_utils",
srcs = [
"lib/ExecutionEngine/RunnerUtils.cpp",
],
hdrs = [
"include/mlir/ExecutionEngine/RunnerUtils.h",
],
includes = ["include"],
deps = [
":mlir_c_runner_utils",
],
)
cc_binary(
name = "mlir-cpu-runner",
srcs = ["tools/mlir-cpu-runner/mlir-cpu-runner.cpp"],
linkopts = ["-ldl"],
deps = [
":AllPassesAndDialectsNoRegistration",
":ExecutionEngineUtils",
":MlirJitRunner",
"@llvm-project//llvm:Support",
],
)
cc_binary(
name = "tools/libcuda-runtime-wrappers.so",
srcs = ["tools/mlir-cuda-runner/cuda-runtime-wrappers.cpp"],
linkshared = True,
deps = [
":mlir_c_runner_utils",
"//third_party/gpus/cuda:cuda_headers",
"//third_party/gpus/cuda:cuda_runtime",
"//third_party/gpus/cuda:libcuda",
"@llvm-project//llvm:Support",
],
)
cc_library(
name = "VulkanRuntime",
srcs = [
"tools/mlir-vulkan-runner/VulkanRuntime.cpp",
],
hdrs = [
"tools/mlir-vulkan-runner/VulkanRuntime.h",
],
deps = [
":IR",
":Pass",
":SPIRVDialect",
":SideEffectInterfaces",
":StandardOps",
":Support",
"@llvm-project//llvm:Support",
"@vulkan_headers",
"@vulkan_sdk//:sdk",
],
)
cc_binary(
name = "tools/libvulkan-runtime-wrappers.so",
srcs = ["tools/mlir-vulkan-runner/vulkan-runtime-wrappers.cpp"],
linkshared = True,
deps = [
":VulkanRuntime",
"@llvm-project//llvm:Support",
],
)
cc_binary(
name = "mlir-cuda-runner",
srcs = ["tools/mlir-cuda-runner/mlir-cuda-runner.cpp"],
data = [":tools/libcuda-runtime-wrappers.so"],
deps = [
":AllPassesAndDialectsNoRegistration",
":ExecutionEngineUtils",
":GPUDialect",
":GPUToGPURuntimeTransforms",
":GPUToNVVMTransforms",
":GPUToROCDLTransforms",
":GPUTransforms",
":IR",
":LLVMDialect",
":MlirJitRunner",
":NVVMDialect",
":Pass",
":StandardToLLVM",
":TargetNVVMIR",
":Transforms",
"//devtools/build/runtime:get_runfiles_dir",
"//third_party/gpus/cuda:cuda_headers",
"//third_party/gpus/cuda:cuda_runtime",
"//third_party/gpus/cuda:libcuda",
"@llvm-project//llvm:Support",
],
)
cc_binary(
name = "mlir-vulkan-runner",
srcs = ["tools/mlir-vulkan-runner/mlir-vulkan-runner.cpp"],
data = [
":tools/libvulkan-runtime-wrappers.so",
"@llvm-project//mlir/test/mlir-cpu-runner:libmlir_runner_utils.so",
],
deps = [
":AllPassesAndDialectsNoRegistration",
":ExecutionEngineUtils",
":GPUToSPIRVTransforms",
":GPUToVulkanTransforms",
":GPUTransforms",
":MlirJitRunner",
":Pass",
":SPIRVDialect",
":StandardToLLVM",
":StandardToSPIRVTransforms",
"@llvm-project//llvm:Support",
],
)
cc_library(
name = "TableGen",
srcs = glob(["lib/TableGen/*.cpp"]),
hdrs = glob(["include/mlir/TableGen/*.h"]),
includes = ["include"],
deps = [
":Support",
"@llvm-project//llvm:Support",
"@llvm-project//llvm:TableGen",
],
)
cc_library(
name = "MlirTableGenMain",
srcs = [
"tools/mlir-tblgen/mlir-tblgen.cpp",
],
includes = ["include"],
deps = [
":Support",
":TableGen",
"@llvm-project//llvm:Support",
"@llvm-project//llvm:TableGen",
"@llvm-project//llvm:config",
],
)
cc_binary(
name = "mlir-tblgen",
srcs = glob([
"tools/mlir-tblgen/*.h",
"tools/mlir-tblgen/*.cpp",
]),
linkopts = [
"-lm",
"-lpthread",
],
deps = [
":MlirTableGenMain",
":Support",
":TableGen",
"@llvm-project//llvm:Support",
"@llvm-project//llvm:TableGen",
"@llvm-project//llvm:config",
],
)
cc_binary(
name = "mlir-linalg-ods-gen",
srcs = glob([
"tools/mlir-linalg-ods-gen/mlir-linalg-ods-gen.cpp",
]),
linkopts = [
"-lm",
"-lpthread",
],
deps = [
":IR",
":Support",
"@llvm-project//llvm:Support",
"@llvm-project//llvm:TableGen",
"@llvm-project//llvm:config",
],
)
## OpenMP dialect
gentbl(
name = "OpenMPOpsIncGen",
strip_include_prefix = "include",
tbl_outs = [
(
"-gen-op-decls",
"include/mlir/Dialect/OpenMP/OpenMPOps.h.inc",
),
(
"-gen-op-defs",
"include/mlir/Dialect/OpenMP/OpenMPOps.cpp.inc",
),
(
"-gen-enum-decls",
"include/mlir/Dialect/OpenMP/OpenMPOpsEnums.h.inc",
),
(
"-gen-enum-defs",
"include/mlir/Dialect/OpenMP/OpenMPOpsEnums.cpp.inc",
),
(
"-gen-dialect-decls",
"include/mlir/Dialect/OpenMP/OpenMPOpsDialect.h.inc",
),
(
"-gen-op-doc",
"g3doc/Dialects/OpenMP/OpenMPOps.md",
),
],
tblgen = ":mlir-tblgen",
td_file = "include/mlir/Dialect/OpenMP/OpenMPOps.td",
td_srcs = [
":OpBaseTdFiles",
],
)
cc_library(
name = "OpenMPDialect",
srcs = glob(
[
"lib/Dialect/OpenMP/IR/*.cpp",
"lib/Dialect/OpenMP/IR/*.h",
],
),
hdrs = glob([
"include/mlir/Dialect/OpenMP/*.h",
]),
includes = ["include"],
deps = [
":IR",
":OpenMPOpsIncGen",
":StandardOps",
"@llvm-project//llvm:Support",
],
)
filegroup(
name = "QuantizationOpsTdFiles",
srcs = [
"include/mlir/Dialect/Quant/QuantOps.td",
"include/mlir/Dialect/Quant/QuantOpsBase.td",
"include/mlir/Interfaces/SideEffectInterfaces.td",
":OpBaseTdFiles",
],
)
## QuantOps dialect
gentbl(
name = "QuantOpsIncGen",
strip_include_prefix = "include",
tbl_outs = [
(
"-gen-op-decls",
"include/mlir/Dialect/Quant/QuantOps.h.inc",
),
(
"-gen-op-defs",
"include/mlir/Dialect/Quant/QuantOps.cpp.inc",
),
(
"-gen-dialect-decls",
"include/mlir/Dialect/Quant/QuantOpsDialect.h.inc",
),
(
"-gen-op-doc",
"g3doc/Dialects/QuantOps/QuantOps.md",
),
],
tblgen = ":mlir-tblgen",
td_file = "include/mlir/Dialect/Quant/QuantOps.td",
td_srcs = [
":QuantizationOpsTdFiles",
],
)
gentbl(
name = "QuantPassIncGen",
strip_include_prefix = "include",
tbl_outs = [
(
"-gen-pass-decls",
"include/mlir/Dialect/Quant/Passes.h.inc",
),
],
tblgen = ":mlir-tblgen",
td_file = "include/mlir/Dialect/Quant/Passes.td",
td_srcs = [
":PassBaseTdFiles",
],
)
cc_library(
name = "QuantOps",
srcs = [
"lib/Dialect/Quant/IR/QuantOps.cpp",
"lib/Dialect/Quant/IR/QuantTypes.cpp",
"lib/Dialect/Quant/IR/TypeDetail.h",
"lib/Dialect/Quant/IR/TypeParser.cpp",
"lib/Dialect/Quant/Transforms/ConvertConst.cpp",
"lib/Dialect/Quant/Transforms/ConvertSimQuant.cpp",
"lib/Dialect/Quant/Transforms/PassDetail.h",
"lib/Dialect/Quant/Utils/FakeQuantSupport.cpp",
"lib/Dialect/Quant/Utils/QuantizeUtils.cpp",
"lib/Dialect/Quant/Utils/UniformSupport.cpp",
],
hdrs = [
"include/mlir/Dialect/Quant/FakeQuantSupport.h",
"include/mlir/Dialect/Quant/Passes.h",
"include/mlir/Dialect/Quant/QuantOps.h",
"include/mlir/Dialect/Quant/QuantTypes.h",
"include/mlir/Dialect/Quant/QuantizeUtils.h",
"include/mlir/Dialect/Quant/UniformSupport.h",
],
includes = ["include"],
deps = [
":IR",
":Pass",
":QuantOpsIncGen",
":QuantPassIncGen",
":SideEffectInterfaces",
":StandardOps",
"@llvm-project//llvm:Support",
],
)
filegroup(
name = "LinalgOpsTdFiles",
srcs = [
"include/mlir/Dialect/Linalg/IR/LinalgBase.td",
"include/mlir/Dialect/Linalg/IR/LinalgOps.td",
"include/mlir/Interfaces/ViewLikeInterface.td",
":AffineOpsTdFiles",
":OpBaseTdFiles",
],
)
gentbl(
name = "LinalgOpsIncGen",
strip_include_prefix = "include",
tbl_outs = [
(
"-gen-op-decls",
"include/mlir/Dialect/Linalg/IR/LinalgOps.h.inc",
),
(
"-gen-op-defs",
"include/mlir/Dialect/Linalg/IR/LinalgOps.cpp.inc",
),
(
"-gen-dialect-decls -dialect=linalg",
"include/mlir/Dialect/Linalg/IR/LinalgOpsDialect.h.inc",
),
],
tblgen = ":mlir-tblgen",
td_file = "include/mlir/Dialect/Linalg/IR/LinalgOps.td",
td_srcs = [
":LinalgOpsTdFiles",
],
)
genlinalg(
name = "LinalgNamedStructuredOpsIncGen",
src = "include/mlir/Dialect/Linalg/IR/LinalgNamedStructuredOpsSpec.tc",
linalg_outs = [
(
"-gen-impl",
"include/mlir/Dialect/Linalg/IR/LinalgNamedStructuredOps.cpp.inc",
),
(
"-gen-ods-decl",
"include/mlir/Dialect/Linalg/IR/LinalgNamedStructuredOps.td",
),
],
linalggen = ":mlir-linalg-ods-gen",
)
filegroup(
name = "LinalgStructuredOpsTdFiles",
srcs = [
"include/mlir/Dialect/Linalg/IR/LinalgNamedStructuredOps.td",
"include/mlir/Dialect/Linalg/IR/LinalgStructuredOps.td",
"include/mlir/Dialect/Linalg/IR/LinalgStructuredOpsInterface.td",
"include/mlir/Interfaces/ViewLikeInterface.td",
":AffineOpsTdFiles",
":LinalgOpsTdFiles",
":OpBaseTdFiles",
],
)
gentbl(
name = "LinalgStructuredOpsIncGen",
strip_include_prefix = "include",
tbl_outs = [
(
"-gen-op-decls",
"include/mlir/Dialect/Linalg/IR/LinalgStructuredOps.h.inc",
),
(
"-gen-op-defs",
"include/mlir/Dialect/Linalg/IR/LinalgStructuredOps.cpp.inc",
),
(
"-gen-op-interface-decls",
"include/mlir/Dialect/Linalg/IR/LinalgStructuredOpsInterfaces.h.inc",
),
(
"-gen-op-interface-defs",
"include/mlir/Dialect/Linalg/IR/LinalgStructuredOpsInterfaces.cpp.inc",
),
],
tblgen = ":mlir-tblgen",
td_file = "include/mlir/Dialect/Linalg/IR/LinalgStructuredOps.td",
td_srcs = [
":LinalgStructuredOpsTdFiles",
"include/mlir/Dialect/Linalg/IR/LinalgNamedStructuredOps.td",
],
)
filegroup(
name = "LinalgDocTdFiles",
srcs = [
"include/mlir/Dialect/Linalg/IR/LinalgDoc.td",
":LinalgOpsTdFiles",
":LinalgStructuredOpsTdFiles",
],
)
gentbl(
name = "LinalgDocIncGen",
strip_include_prefix = "include",
tbl_outs = [
(
"-gen-op-doc",
"g3doc/Dialects/Linalg/LinalgOps.md",
),
],
tblgen = ":mlir-tblgen",
td_file = "include/mlir/Dialect/Linalg/IR/LinalgDoc.td",
td_srcs = [
":LinalgDocTdFiles",
],
)
cc_library(
name = "LinalgToLLVM",
srcs = glob([
"lib/Conversion/LinalgToLLVM/*.cpp",
"lib/Conversion/LinalgToLLVM/*.h",
]) + ["lib/Conversion/PassDetail.h"],
hdrs = glob([
"include/mlir/Conversion/LinalgToLLVM/*.h",
]),
includes = ["include"],
deps = [
":AffineToStandard",
":Analysis",
":ConversionPassIncGen",
":EDSC",
":IR",
":LLVMDialect",
":LinalgOps",
":LinalgTransforms",
":Pass",
":SCFToStandard",
":StandardOps",
":StandardToLLVM",
":Support",
":Transforms",
":VectorToLLVM",
":VectorToSCF",
"@llvm-project//llvm:Core",
"@llvm-project//llvm:Support",
],
)
cc_library(
name = "LinalgToStandard",
srcs = glob([
"lib/Conversion/LinalgToStandard/*.cpp",
"lib/Conversion/LinalgToStandard/*.h",
]) + ["lib/Conversion/PassDetail.h"],
hdrs = glob([
"include/mlir/Conversion/LinalgToStandard/*.h",
]),
includes = ["include"],
deps = [
":Affine",
":ConversionPassIncGen",
":IR",
":LinalgOps",
":Pass",
":SCFDialect",
":StandardOps",
":Support",
":Transforms",
"@llvm-project//llvm:Core",
"@llvm-project//llvm:Support",
],
)
cc_library(
name = "LinalgToSPIRV",
srcs = glob([
"lib/Conversion/LinalgToSPIRV/*.cpp",
"lib/Conversion/LinalgToSPIRV/*.h",
]) + ["lib/Conversion/PassDetail.h"],
hdrs = glob([
"include/mlir/Conversion/LinalgToSPIRV/*.h",
]),
includes = ["include"],
deps = [
":ConversionPassIncGen",
":DialectUtils",
":IR",
":LinalgOps",
":LinalgTransforms",
":Pass",
":SPIRVDialect",
":SPIRVLowering",
":StandardOps",
],
)
cc_library(
name = "LinalgOps",
srcs = [
"lib/Dialect/Linalg/IR/LinalgOps.cpp",
"lib/Dialect/Linalg/IR/LinalgTypes.cpp",
],
hdrs = [
"include/mlir/Dialect/Linalg/EDSC/Intrinsics.h",
"include/mlir/Dialect/Linalg/IR/LinalgOps.h",
"include/mlir/Dialect/Linalg/IR/LinalgTraits.h",
"include/mlir/Dialect/Linalg/IR/LinalgTypes.h",
],
includes = ["include"],
deps = [
":Affine",
":DialectUtils",
":EDSC",
":IR",
":LinalgNamedStructuredOpsIncGen",
":LinalgOpsIncGen",
":LinalgStructuredOpsIncGen",
":Parser",
":SideEffectInterfaces",
":StandardOps",
":Support",
":ViewLikeInterface",
"@llvm-project//llvm:Support",
],
)
gentbl(
name = "LinalgPassIncGen",
strip_include_prefix = "include",
tbl_outs = [
(
"-gen-pass-decls",
"include/mlir/Dialect/Linalg/Passes.h.inc",
),
],
tblgen = ":mlir-tblgen",
td_file = "include/mlir/Dialect/Linalg/Passes.td",
td_srcs = [
":PassBaseTdFiles",
],
)
cc_library(
name = "LinalgTransforms",
srcs = glob([
"lib/Dialect/Linalg/Transforms/*.cpp",
"lib/Dialect/Linalg/Transforms/*.h",
]) + [
"lib/Dialect/Linalg/Analysis/DependenceAnalysis.cpp",
"lib/Dialect/Linalg/EDSC/Builders.cpp",
"lib/Dialect/Linalg/Utils/Utils.cpp",
],
hdrs = [
"include/mlir/Dialect/Linalg/Analysis/DependenceAnalysis.h",
"include/mlir/Dialect/Linalg/EDSC/Builders.h",
"include/mlir/Dialect/Linalg/EDSC/FoldedIntrinsics.h",
"include/mlir/Dialect/Linalg/Passes.h",
"include/mlir/Dialect/Linalg/Transforms/Hoisting.h",
"include/mlir/Dialect/Linalg/Transforms/Transforms.h",
"include/mlir/Dialect/Linalg/Utils/Utils.h",
],
includes = ["include"],
deps = [
":Affine",
":AffineToStandard",
":Analysis",
":DialectUtils",
":EDSC",
":IR",
":LLVMDialect",
":LinalgOps",
":LinalgPassIncGen",
":LinalgStructuredOpsIncGen",
":Pass",
":SCFDialect",
":SCFToStandard",
":SCFTransforms",
":StandardOps",
":StandardToLLVM",
":Support",
":TransformUtils",
":Transforms",
":TransformsPassIncGen",
":VectorOps",
"@llvm-project//llvm:Core",
"@llvm-project//llvm:Support",
],
)
filegroup(
name = "VectorOpsTdFiles",
srcs = [
"include/mlir/Dialect/Vector/VectorOps.td",
":AffineOpsTdFiles",
":OpBaseTdFiles",
],
)
gentbl(
name = "VectorOpsIncGen",
strip_include_prefix = "include",
tbl_outs = [
(
"-gen-op-decls",
"include/mlir/Dialect/Vector/VectorOps.h.inc",
),
(
"-gen-op-defs",
"include/mlir/Dialect/Vector/VectorOps.cpp.inc",
),
(
"-gen-dialect-decls -dialect=vector",
"include/mlir/Dialect/Vector/VectorOpsDialect.h.inc",
),
(
"-gen-op-doc",
"g3doc/Dialects/Vector/VectorOps.md",
),
],
tblgen = ":mlir-tblgen",
td_file = "include/mlir/Dialect/Vector/VectorOps.td",
td_srcs = [
":VectorOpsTdFiles",
],
)
filegroup(
name = "VectorTransformPatternsTdFiles",
srcs = [
"include/mlir/Dialect/Vector/VectorTransformPatterns.td",
":AffineOpsTdFiles",
":LinalgOpsTdFiles",
":LinalgStructuredOpsTdFiles",
":OpBaseTdFiles",
":StdOpsTdFiles",
":VectorOpsTdFiles",
],
)
gentbl(
name = "VectorTransformPatternsIncGen",
tbl_outs = [
(
"-gen-rewriters",
"include/mlir/Dialect/Vector/VectorTransformPatterns.h.inc",
),
],
tblgen = ":mlir-tblgen",
td_file = "include/mlir/Dialect/Vector/VectorTransformPatterns.td",
td_srcs = [
":VectorTransformPatternsTdFiles",
],
)
cc_library(
name = "VectorToLLVM",
srcs = glob([
"lib/Conversion/VectorToLLVM/*.cpp",
"lib/Conversion/VectorToLLVM/*.h",
]) + ["lib/Conversion/PassDetail.h"],
hdrs = glob([
"include/mlir/Conversion/VectorToLLVM/*.h",
]),
includes = ["include"],
deps = [
":ConversionPassIncGen",
":DialectUtils",
":EDSC",
":IR",
":LLVMDialect",
":Pass",
":StandardOps",
":StandardToLLVM",
":Support",
":Transforms",
":VectorOps",
"@llvm-project//llvm:Core",
"@llvm-project//llvm:Support",
],
)
cc_library(
name = "VectorToSCF",
srcs = glob([
"lib/Conversion/VectorToSCF/*.cpp",
"lib/Conversion/VectorToSCF/*.h",
]) + ["lib/Conversion/PassDetail.h"],
hdrs = glob([
"include/mlir/Conversion/VectorToSCF/*.h",
]),
includes = ["include"],
deps = [
":Affine",
":ConversionPassIncGen",
":EDSC",
":IR",
":LLVMDialect",
":Pass",
":SCFDialect",
":StandardOps",
":StandardToLLVM",
":Support",
":Transforms",
":VectorOps",
"@llvm-project//llvm:Core",
"@llvm-project//llvm:Support",
],
)
# To reference all tablegen files here when checking for updates to them.
filegroup(
name = "TdFiles",
srcs = glob(["**/*.td"]),
)
exports_files(
[
"include/mlir/Interfaces/CallInterfaces.h",
"include/mlir/Interfaces/CallInterfaces.td",
"include/mlir/Interfaces/ControlFlowInterfaces.h",
"include/mlir/Interfaces/ControlFlowInterfaces.td",
"include/mlir/Interfaces/SideEffectInterfaces.td",
"include/mlir/Interfaces/ViewLikeInterface.td",
"include/mlir/Dialect/LLVMIR/LLVMOpBase.td",
"include/mlir/Dialect/StandardOps/IR/Ops.td",
"include/mlir/IR/OpAsmInterface.td",
"include/mlir/IR/OpBase.td",
"include/mlir/IR/SymbolInterfaces.td",
"include/mlir/Transforms/InliningUtils.h",
],
visibility = [":friends"],
)
exports_files(
[
"include/mlir/Interfaces/InferTypeOpInterface.td",
"include/mlir/Interfaces/LoopLikeInterface.td",
],
visibility = [":friends"],
)