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

Integrate LLVM at llvm/llvm-project@499bce3aba

Browse Source 
Updates LLVM usage to match
[499bce3abab8](https://github.com/llvm/llvm-project/commit/499bce3abab8)

PiperOrigin-RevId: 342921326
Change-Id: Ide3b8a2ba3027d1277a25540948d280d115f5d1b
This commit is contained in:
A. Unique TensorFlower 2020-11-17 12:29:09 -08:00 committed by TensorFlower Gardener
parent 5186e8295c
commit 4fe5106a8f
5 changed files with 14 additions and 9 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
tensorflow/compiler/mlir/hlo/BUILD
View File

@ -664,6 +664,7 @@ cc_library(
"@llvm-project//mlir:Shape", "@llvm-project//mlir:Shape",
"@llvm-project//mlir:ShapeTransforms", "@llvm-project//mlir:ShapeTransforms",
"@llvm-project//mlir:StandardOps", "@llvm-project//mlir:StandardOps",
"@llvm-project//mlir:StandardOpsTransforms",
"@llvm-project//mlir:Support", "@llvm-project//mlir:Support",
"@llvm-project//mlir:Transforms", "@llvm-project//mlir:Transforms",
], ],

6
tensorflow/compiler/mlir/hlo/lib/Dialect/mhlo/transforms/hlo_legalize_to_lhlo.cc
View File

@ -23,6 +23,7 @@ limitations under the License.
#include "mlir/Dialect/Shape/IR/Shape.h" #include "mlir/Dialect/Shape/IR/Shape.h"
#include "mlir/Dialect/Shape/Transforms/Passes.h" #include "mlir/Dialect/Shape/Transforms/Passes.h"
#include "mlir/Dialect/StandardOps/IR/Ops.h" #include "mlir/Dialect/StandardOps/IR/Ops.h"
#include "mlir/Dialect/StandardOps/Transforms/FuncConversions.h"
#include "mlir/IR/AffineMap.h" #include "mlir/IR/AffineMap.h"
#include "mlir/IR/Attributes.h" #include "mlir/IR/Attributes.h"
#include "mlir/IR/BlockAndValueMapping.h" #include "mlir/IR/BlockAndValueMapping.h"
@ -578,9 +579,8 @@ struct HloLegalizeToLhlo
}); });
populateHLOToLHLOConversionPattern(&context, &converter, &patterns); populateHLOToLHLOConversionPattern(&context, &converter, &patterns);
populateWithBufferizeOpConversionPatterns<mlir::ReturnOp, mlir::ReturnOp, populateFuncOpTypeConversionPattern(patterns, &context, converter);
lmhlo::CopyOp>( populateCallOpTypeConversionPattern(patterns, &context, converter);
&context, converter, patterns);
populateShapeStructuralTypeConversionsAndLegality(&context, converter, populateShapeStructuralTypeConversionsAndLegality(&context, converter,
patterns, target); patterns, target);
if (failed(applyPartialConversion(getOperation(), target, if (failed(applyPartialConversion(getOperation(), target,

10
tensorflow/compiler/mlir/tools/kernel_gen/transforms/bufferize_pass.cc
View File

@ -23,6 +23,8 @@ limitations under the License.
#include "mlir/Dialect/Shape/IR/Shape.h" // from @llvm-project #include "mlir/Dialect/Shape/IR/Shape.h" // from @llvm-project
#include "mlir/Dialect/Shape/Transforms/Passes.h" // from @llvm-project #include "mlir/Dialect/Shape/Transforms/Passes.h" // from @llvm-project
#include "mlir/Dialect/StandardOps/IR/Ops.h" // from @llvm-project #include "mlir/Dialect/StandardOps/IR/Ops.h" // from @llvm-project
#include "mlir/Dialect/StandardOps/Transforms/DecomposeCallGraphTypes.h" // from @llvm-project
#include "mlir/Dialect/StandardOps/Transforms/FuncConversions.h" // from @llvm-project
#include "mlir/Dialect/StandardOps/Transforms/Passes.h" // from @llvm-project #include "mlir/Dialect/StandardOps/Transforms/Passes.h" // from @llvm-project
#include "mlir/IR/Function.h" // from @llvm-project #include "mlir/IR/Function.h" // from @llvm-project
#include "mlir/IR/MLIRContext.h" // from @llvm-project #include "mlir/IR/MLIRContext.h" // from @llvm-project
@ -107,9 +109,11 @@ struct BufferizePass : public BufferizePassBase<BufferizePass> {
OwningRewritePatternList patterns; OwningRewritePatternList patterns;
mhlo::populateHLOToLHLOConversionPattern(&context, &converter, &patterns); mhlo::populateHLOToLHLOConversionPattern(&context, &converter, &patterns);
populateWithBufferizeOpConversionPatterns<ReturnOp, ReturnOp, populateFuncOpTypeConversionPattern(patterns, &context, converter);
lmhlo::CopyOp>( populateCallOpTypeConversionPattern(patterns, &context, converter);
&context, converter, patterns); ValueDecomposer decomposer;
populateDecomposeCallGraphTypesPatterns(&context, converter, decomposer,
patterns);
populateStdBufferizePatterns(&context, converter, patterns); populateStdBufferizePatterns(&context, converter, patterns);
populateExtraStdBufferizePattern(&context, &converter, &patterns); populateExtraStdBufferizePattern(&context, &converter, &patterns);
populateShapeStructuralTypeConversionsAndLegality(&context, converter, populateShapeStructuralTypeConversionsAndLegality(&context, converter,

4
tensorflow/workspace.bzl
View File

@ -686,8 +686,8 @@ def tf_repositories(path_prefix = "", tf_repo_name = ""):
) )
# Check out LLVM and MLIR from llvm-project. # Check out LLVM and MLIR from llvm-project.
LLVM_COMMIT = "65f3e121fe4f80e2b178fa157ef1346043555b4b" LLVM_COMMIT = "499bce3abab8a362b9b4197944bd40b826c736c4"
LLVM_SHA256 = "483154bacee5852087821850a64d43ebefd088a2e19f3eeb43391cafea3749d8" LLVM_SHA256 = "a154965dfeb2b5963acc2193bc334ce90b314acbe48430ba310d8a7c7a20de8b"
LLVM_URLS = [ LLVM_URLS = [
"https://storage.googleapis.com/mirror.tensorflow.org/github.com/llvm/llvm-project/archive/{commit}.tar.gz".format(commit = LLVM_COMMIT), "https://storage.googleapis.com/mirror.tensorflow.org/github.com/llvm/llvm-project/archive/{commit}.tar.gz".format(commit = LLVM_COMMIT),
"https://github.com/llvm/llvm-project/archive/{commit}.tar.gz".format(commit = LLVM_COMMIT), "https://github.com/llvm/llvm-project/archive/{commit}.tar.gz".format(commit = LLVM_COMMIT),

2
third_party/mlir/BUILD vendored
View File

@ -3699,7 +3699,7 @@ gentbl(
"include/mlir/Dialect/OpenMP/OpenMPOpsEnums.cpp.inc", "include/mlir/Dialect/OpenMP/OpenMPOpsEnums.cpp.inc",
), ),
( (
"-gen-dialect-decls", "-gen-dialect-decls -dialect=omp",
"include/mlir/Dialect/OpenMP/OpenMPOpsDialect.h.inc", "include/mlir/Dialect/OpenMP/OpenMPOpsDialect.h.inc",
), ),
( (