experiments/STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Internal change

Browse Source 
PiperOrigin-RevId: 303029359
Change-Id: Ia17a786ded245979c2e4a448405a22ffcfe57da1
This commit is contained in:
A. Unique TensorFlower 2020-03-25 20:25:09 -07:00 committed by TensorFlower Gardener
parent 81415e2bfc
commit 2d055a4226
1 changed files with 39 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

46
third_party/mlir/BUILD vendored
View File

@ -846,6 +846,7 @@ cc_library(
filegroup( filegroup(
name = "GPUOpsTdFiles", name = "GPUOpsTdFiles",
srcs = [ srcs = [
"include/mlir/Dialect/GPU/GPUBase.td",
"include/mlir/Dialect/GPU/GPUOps.td", "include/mlir/Dialect/GPU/GPUOps.td",
"include/mlir/Dialect/LLVMIR/LLVMOpBase.td", "include/mlir/Dialect/LLVMIR/LLVMOpBase.td",
"include/mlir/Interfaces/SideEffects.td", "include/mlir/Interfaces/SideEffects.td",
@ -853,6 +854,35 @@ filegroup(
], ],
) )
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( gentbl(
name = "GPUOpsIncGen", name = "GPUOpsIncGen",
strip_include_prefix = "include", strip_include_prefix = "include",
@ -886,7 +916,7 @@ cc_library(
], ],
), ),
hdrs = glob([ hdrs = glob([
"include/mlir/Dialect/GPU/*.h", "include/mlir/Dialect/GPU/GPUDialect.h",
]), ]),
includes = ["include"], includes = ["include"],
deps = [ deps = [
@ -908,6 +938,8 @@ cc_library(
], ],
), ),
hdrs = [ hdrs = [
"include/mlir/Dialect/GPU/MemoryPromotion.h",
"include/mlir/Dialect/GPU/ParallelLoopMapper.h",
"include/mlir/Dialect/GPU/Passes.h", "include/mlir/Dialect/GPU/Passes.h",
"include/mlir/Dialect/GPU/Utils.h", "include/mlir/Dialect/GPU/Utils.h",
], ],
@ -917,10 +949,12 @@ cc_library(
":GPUDialect", ":GPUDialect",
":IR", ":IR",
":LoopOps", ":LoopOps",
":ParallelLoopMapperAttrGen",
":Pass", ":Pass",
":StandardOps", ":StandardOps",
":Support", ":Support",
":Transforms", ":Transforms",
"@llvm-project//llvm:support",
], ],
) )
@ -982,6 +1016,7 @@ cc_library(
":GPUCommonTransforms", ":GPUCommonTransforms",
":GPUDialect", ":GPUDialect",
":GPUToNVVMGen", ":GPUToNVVMGen",
":GPUTransforms",
":IR", ":IR",
":LLVMTransforms", ":LLVMTransforms",
":NVVMDialect", ":NVVMDialect",
@ -1733,17 +1768,14 @@ cc_library(
cc_library( cc_library(
name = "LoopsToGPU", name = "LoopsToGPU",
srcs = [ srcs = ["lib/Conversion/LoopsToGPU/LoopsToGPU.cpp"],
"lib/Conversion/LoopsToGPU/LoopsToGPU.cpp", hdrs = ["include/mlir/Conversion/LoopsToGPU/LoopsToGPU.h"],
],
hdrs = [
"include/mlir/Conversion/LoopsToGPU/LoopsToGPU.h",
],
includes = ["include"], includes = ["include"],
deps = [ deps = [
":Affine", ":Affine",
":AffineToStandardTransforms", ":AffineToStandardTransforms",
":GPUDialect", ":GPUDialect",
":GPUTransforms",
":IR", ":IR",
":LoopOps", ":LoopOps",
":Pass", ":Pass",