diff options
| author |  A. Unique TensorFlower <gardener@tensorflow.org> | 2017-08-04 18:45:25 -0700 |
|---|---|---|
| committer | TensorFlower Gardener <gardener@tensorflow.org> | 2017-08-04 18:48:53 -0700 |
| commit | 69db15cc49f84baaff27a19d792102876e811c97 (patch) | |
| tree | efbeea278a1587f6ab2cfee135a0b953e8da6f6a | |
| parent | 1343c02da2b19bc4430a1c1b80592d8f6f098589 (diff) | |
Include more appropriate LLVM headers
PiperOrigin-RevId: 164336354
| -rw-r--r-- | tensorflow/compiler/xla/service/gpu/llvm_gpu_backend/BUILD | 6 | ||||
| -rw-r--r-- | tensorflow/compiler/xla/service/gpu/llvm_gpu_backend/gpu_backend_lib.cc | 7 |
2 files changed, 5 insertions, 8 deletions
diff --git a/tensorflow/compiler/xla/service/gpu/llvm_gpu_backend/BUILD b/tensorflow/compiler/xla/service/gpu/llvm_gpu_backend/BUILD index 1d1e5bee54..e362eb1802 100644 --- a/tensorflow/compiler/xla/service/gpu/llvm_gpu_backend/BUILD +++ b/tensorflow/compiler/xla/service/gpu/llvm_gpu_backend/BUILD @@ -2,10 +2,7 @@ licenses(["notice"]) # Apache 2.0 package( default_visibility = [":friends"], - features = [ - "-parse_headers", - "no_layering_check", - ], + features = ["-parse_headers"], ) package_group( @@ -49,6 +46,7 @@ cc_library( "@llvm//:mc", "@llvm//:nvptx_code_gen", "@llvm//:objc_arc", + "@llvm//:scalar", "@llvm//:support", "@llvm//:target", "@llvm//:transform_utils", diff --git a/tensorflow/compiler/xla/service/gpu/llvm_gpu_backend/gpu_backend_lib.cc b/tensorflow/compiler/xla/service/gpu/llvm_gpu_backend/gpu_backend_lib.cc index 217a707fcb..2a999f52f0 100644 --- a/tensorflow/compiler/xla/service/gpu/llvm_gpu_backend/gpu_backend_lib.cc +++ b/tensorflow/compiler/xla/service/gpu/llvm_gpu_backend/gpu_backend_lib.cc @@ -38,8 +38,7 @@ limitations under the License. #include "llvm/IR/LLVMContext.h" #include "llvm/IR/LegacyPassManager.h" #include "llvm/IR/Module.h" -#include "llvm/LinkAllIR.h" -#include "llvm/LinkAllPasses.h" +#include "llvm/IR/Verifier.h" #include "llvm/Linker/Linker.h" #include "llvm/PassRegistry.h" #include "llvm/Support/CommandLine.h" @@ -51,9 +50,9 @@ limitations under the License. #include "llvm/Target/TargetMachine.h" #include "llvm/Transforms/IPO.h" #include "llvm/Transforms/IPO/AlwaysInliner.h" -#include "llvm/Transforms/IPO/PassManagerBuilder.h" - #include "llvm/Transforms/IPO/Internalize.h" +#include "llvm/Transforms/IPO/PassManagerBuilder.h" +#include "llvm/Transforms/Scalar.h" #include "tensorflow/compiler/xla/types.h" #include "tensorflow/core/lib/core/stringpiece.h" #include "tensorflow/core/lib/io/path.h" |