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

Integrate LLVM at https://github.com/llvm/llvm-project/commit/c2171457e281

Browse Source 
PiperOrigin-RevId: 306950518
Change-Id: I6da501ea9c226bfeb35f8854b6a12dc7a42938df
This commit is contained in:
Benjamin Kramer 2020-04-16 17:04:19 -07:00 committed by TensorFlower Gardener
parent a64755b44f
commit 6b0ab72dfc
3 changed files with 7 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
tensorflow/compiler/mlir/xla/transforms/xla_legalize_to_linalg.cc
View File

@ -134,7 +134,7 @@ class PointwiseToLinalgConverter : public OpConversionPattern<OpTy> {
rewriter.getI64IntegerAttr(bodyResultTypes.size()), // args_out
rewriter.getArrayAttr(indexingMaps),
GetNParallelLoopsAttrs(nloops, &rewriter),
/*doc=*/nullptr, /*fun=*/nullptr, /*library_call=*/nullptr);
/*doc=*/nullptr, /*library_call=*/nullptr);
// Add a block to the region.
auto* region = &linalgOp.region();
@ -218,7 +218,7 @@ class DataMovementOpConverter : public OpConversionPattern<OpTy> {
loc, isLHLO ? ArrayRef<Type>{} : resultType, args,
rewriter.getI64IntegerAttr(1), rewriter.getI64IntegerAttr(1),
indexingMapsAttr, GetNParallelLoopsAttrs(nloops, &rewriter),
/*doc=*/nullptr, /*fun=*/nullptr, /*library_call=*/nullptr);
/*doc=*/nullptr, /*library_call=*/nullptr);
auto* region = &linalgOp.region();
auto* block = rewriter.createBlock(region, region->end());
@ -400,7 +400,7 @@ class IotaConverter : public OpConversionPattern<xla_lhlo::IotaOp> {
rewriter.getI64IntegerAttr(1), // args_out
rewriter.getArrayAttr(indexingMaps),
GetNParallelLoopsAttrs(nloops, &rewriter),
/*doc=*/nullptr, /*fun=*/nullptr, /*library_call=*/nullptr);
/*doc=*/nullptr, /*library_call=*/nullptr);
// Add a block to the region.
auto* region = &linalgOp.region();

4
tensorflow/workspace.bzl
View File

@ -658,8 +658,8 @@ def tf_repositories(path_prefix = "", tf_repo_name = ""):
)
# Check out LLVM and MLIR from llvm-project.
LLVM_COMMIT = "3ee1ec0b9dd6ee2350f39ae8a418bf3ce28d06cf"
LLVM_SHA256 = "473eae82b2c9f8bace1b86018b809a1565723d0f9affaeb88d3d95b685062746"
LLVM_COMMIT = "c2171457e28130cbb61fae74ffe0d53547a7a505"
LLVM_SHA256 = "79a04f64fa33967034e6dbd1c42a7af30b975a289ab30c2d8209fa2aef0d2020"
LLVM_URLS = [
"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),

2
third_party/mlir/BUILD vendored
View File

@ -901,6 +901,8 @@ cc_library(
":SideEffects",
":Support",
"@llvm-project//llvm:asm_parser",
"@llvm-project//llvm:bit_reader",
"@llvm-project//llvm:bit_writer",
"@llvm-project//llvm:core",
"@llvm-project//llvm:support",
],