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/7af0c8559b6d

Browse Source 
PiperOrigin-RevId: 311914136
Change-Id: I8a04df09178d9dfce79c13c8d14daf4f69048dee
This commit is contained in:
A. Unique TensorFlower 2020-05-16 16:12:32 -07:00 committed by TensorFlower Gardener
parent 766f2968fc
commit bf639d750b
11 changed files with 15 additions and 19 deletions
tensorflow/compiler
mlir
lite/ir
tfl_ops.h
tensorflow
ir
control_flow_ops.htf_ops.h
transforms
constant_fold.cc
tfjs/ir
tfjs_ops.h
xla/ir
chlo_ops.hhlo_ops.hlhlo_ops.h
xla/service
cpu
ir_emitter.cc
gpu
ir_emitter.cc
third_party/mlir
BUILD

2
tensorflow/compiler/mlir/lite/ir/tfl_ops.h
View File

@ -27,7 +27,7 @@ limitations under the License.
#include "mlir/IR/StandardTypes.h" // from @llvm-project
#include "mlir/Interfaces/DerivedAttributeOpInterface.h" // from @llvm-project
#include "mlir/Interfaces/LoopLikeInterface.h" // from @llvm-project
#include "mlir/Interfaces/SideEffects.h" // from @llvm-project
#include "mlir/Interfaces/SideEffectInterfaces.h" // from @llvm-project
#include "mlir/Support/LLVM.h" // from @llvm-project
#include "tensorflow/compiler/mlir/lite/quantization/quantization_traits.h"
#include "tensorflow/lite/schema/schema_generated.h"

2
tensorflow/compiler/mlir/tensorflow/ir/control_flow_ops.h
View File

@ -26,7 +26,7 @@ limitations under the License.
#include "mlir/IR/Dialect.h" // from @llvm-project
#include "mlir/IR/OpDefinition.h" // from @llvm-project
#include "mlir/IR/Types.h" // from @llvm-project
#include "mlir/Interfaces/SideEffects.h" // from @llvm-project
#include "mlir/Interfaces/SideEffectInterfaces.h" // from @llvm-project
namespace mlir {
namespace TFControlFlow {

2
tensorflow/compiler/mlir/tensorflow/ir/tf_ops.h
View File

@ -31,7 +31,7 @@ limitations under the License.
#include "mlir/Interfaces/CallInterfaces.h" // from @llvm-project
#include "mlir/Interfaces/DerivedAttributeOpInterface.h" // from @llvm-project
#include "mlir/Interfaces/InferTypeOpInterface.h" // from @llvm-project
#include "mlir/Interfaces/SideEffects.h" // from @llvm-project
#include "mlir/Interfaces/SideEffectInterfaces.h" // from @llvm-project
#include "tensorflow/compiler/mlir/tensorflow/ir/tf_attributes.h"
#include "tensorflow/compiler/mlir/tensorflow/ir/tf_structs.h"
#include "tensorflow/compiler/mlir/tensorflow/ir/tf_traits.h"

2
tensorflow/compiler/mlir/tensorflow/transforms/constant_fold.cc
View File

@ -17,7 +17,7 @@ limitations under the License.
#include <algorithm>
#include "mlir/Interfaces/SideEffects.h" // from @llvm-project
#include "mlir/Interfaces/SideEffectInterfaces.h" // from @llvm-project
#include "mlir/Support/LogicalResult.h" // from @llvm-project
#include "tensorflow/c/eager/c_api.h"
#include "tensorflow/c/tf_status.h"

2
tensorflow/compiler/mlir/tfjs/ir/tfjs_ops.h
View File

@ -26,7 +26,7 @@ limitations under the License.
#include "mlir/IR/Dialect.h" // from @llvm-project
#include "mlir/IR/OpImplementation.h" // from @llvm-project
#include "mlir/IR/StandardTypes.h" // from @llvm-project
#include "mlir/Interfaces/SideEffects.h" // from @llvm-project
#include "mlir/Interfaces/SideEffectInterfaces.h" // from @llvm-project
#include "mlir/Support/LLVM.h" // from @llvm-project
namespace mlir {

2
tensorflow/compiler/mlir/xla/ir/chlo_ops.h
View File

@ -25,7 +25,7 @@ limitations under the License.
#include "mlir/IR/StandardTypes.h" // from @llvm-project
#include "mlir/IR/Types.h" // from @llvm-project
#include "mlir/Interfaces/InferTypeOpInterface.h" // from @llvm-project
#include "mlir/Interfaces/SideEffects.h" // from @llvm-project
#include "mlir/Interfaces/SideEffectInterfaces.h" // from @llvm-project
namespace mlir {
namespace xla_chlo {

2
tensorflow/compiler/mlir/xla/ir/hlo_ops.h
View File

@ -29,7 +29,7 @@ limitations under the License.
#include "mlir/IR/StandardTypes.h" // from @llvm-project
#include "mlir/IR/Types.h" // from @llvm-project
#include "mlir/Interfaces/InferTypeOpInterface.h" // from @llvm-project
#include "mlir/Interfaces/SideEffects.h" // from @llvm-project
#include "mlir/Interfaces/SideEffectInterfaces.h" // from @llvm-project
namespace mlir {
class OpBuilder;

2
tensorflow/compiler/mlir/xla/ir/lhlo_ops.h
View File

@ -27,7 +27,7 @@ limitations under the License.
#include "mlir/IR/Operation.h" // from @llvm-project
#include "mlir/IR/StandardTypes.h" // from @llvm-project
#include "mlir/IR/Types.h" // from @llvm-project
#include "mlir/Interfaces/SideEffects.h" // from @llvm-project
#include "mlir/Interfaces/SideEffectInterfaces.h" // from @llvm-project
namespace mlir {
class OpBuilder;

5
tensorflow/compiler/xla/service/cpu/ir_emitter.cc
View File

@ -2875,9 +2875,8 @@ Status IrEmitter::HandleRngGetAndUpdateState(HloInstruction* rng_state) {
old_state->getType()->getScalarType(),
address->getType()->getPointerAddressSpace()));
llvm::StoreInst* store = Store(old_state, address);
store->setAlignment(
llvm::MaybeAlign(IrEmitter::MinimumAlignmentForPrimitiveType(
rng_state->shape().element_type())));
store->setAlignment(llvm::Align(IrEmitter::MinimumAlignmentForPrimitiveType(
rng_state->shape().element_type())));
return Status::OK();
}

2
tensorflow/compiler/xla/service/gpu/ir_emitter.cc
View File

@ -222,7 +222,7 @@ bool IrEmitter::MaybeEmitDirectAtomicOperation(
// Derive a minimum alignment from the type. The optimizer can increase it
// later.
store->setAlignment(
llvm::MaybeAlign(ShapeUtil::ByteSizeOfPrimitiveType(element_type)));
llvm::Align(ShapeUtil::ByteSizeOfPrimitiveType(element_type)));
return true;
}

11
third_party/mlir/BUILD vendored
View File

@ -788,9 +788,6 @@ cc_library(
"lib/Support/*.h",
],
exclude = [
# TODO(herhut): Move JitRunner out of Support so that Support does not
# depend on dialect.
"lib/Support/JitRunner.cpp",
# TODO(jpienaar): Move this out, else Support depends on Analysis/
"lib/Support/MlirOptMain.cpp",
],
@ -2232,10 +2229,10 @@ gentbl(
cc_library(
name = "SideEffects",
srcs = [
"lib/Interfaces/SideEffects.cpp",
"lib/Interfaces/SideEffectInterfaces.cpp",
],
hdrs = [
"include/mlir/Interfaces/SideEffects.h",
"include/mlir/Interfaces/SideEffectInterfaces.h",
],
includes = ["include"],
deps = [
@ -2621,8 +2618,8 @@ cc_binary(
cc_library(
name = "MlirJitRunner",
srcs = ["lib/Support/JitRunner.cpp"],
hdrs = ["include/mlir/Support/JitRunner.h"],
srcs = ["lib/ExecutionEngine/JitRunner.cpp"],
hdrs = ["include/mlir/ExecutionEngine/JitRunner.h"],
includes = ["include"],
deps = [
":AllPassesAndDialectsNoRegistration",