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

Adapt code to LLVM API updates.

Browse Source 
PiperOrigin-RevId: 280431812
Change-Id: I52ed3b778a822c8a7c63791a6ad25d2c473b5d89
This commit is contained in:
A. Unique TensorFlower 2019-11-14 08:15:25 -08:00 committed by TensorFlower Gardener
parent 675826bd90
commit 4cd0a9a9b0
3 changed files with 6 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
tensorflow/compiler/aot/embedded_protocol_buffers.cc
View File

@ -80,9 +80,8 @@ static StatusOr<string> CodegenModule(llvm::TargetMachine* target_machine,
llvm::raw_svector_ostream ostream(stream_buffer);
llvm::legacy::PassManager codegen_passes;
if (target_machine->addPassesToEmitFile(
codegen_passes, ostream, nullptr,
llvm::TargetMachine::CGFT_ObjectFile)) {
if (target_machine->addPassesToEmitFile(codegen_passes, ostream, nullptr,
llvm::CGFT_ObjectFile)) {
return xla::InternalError(
"Could not create pass pipeline to generate object file");
}

5
tensorflow/compiler/xla/service/gpu/llvm_gpu_backend/gpu_backend_lib.cc
View File

@ -36,6 +36,7 @@ limitations under the License.
#include "llvm/IR/LegacyPassManager.h"
#include "llvm/IR/Module.h"
#include "llvm/IR/Verifier.h"
#include "llvm/InitializePasses.h"
#include "llvm/Linker/Linker.h"
#include "llvm/PassRegistry.h"
#include "llvm/Support/CommandLine.h"
@ -225,7 +226,7 @@ string EmitModuleToPTX(Module* module, llvm::TargetMachine* target_machine) {
llvm::Triple(module->getTargetTriple())));
target_machine->addPassesToEmitFile(codegen_passes, pstream, nullptr,
llvm::TargetMachine::CGFT_AssemblyFile);
llvm::CGFT_AssemblyFile);
codegen_passes.run(*module);
}
@ -613,7 +614,7 @@ StatusOr<std::vector<uint8>> EmitModuleToHsaco(
new llvm::raw_fd_ostream(isabin_path, ec, llvm::sys::fs::F_Text));
module->setDataLayout(target_machine->createDataLayout());
target_machine->addPassesToEmitFile(codegen_passes, *isabin_fs, nullptr,
llvm::TargetMachine::CGFT_ObjectFile);
llvm::CGFT_ObjectFile);
codegen_passes.run(*module);
isabin_fs->flush();

2
third_party/mlir/lib/Conversion/GPUToCUDA/ConvertKernelFuncToCubin.cpp vendored
View File

@ -117,7 +117,7 @@ std::string GpuKernelToCubinPass::translateModuleToPtx(
llvm::buffer_ostream pstream(stream);
llvm::legacy::PassManager codegen_passes;
target_machine.addPassesToEmitFile(codegen_passes, pstream, nullptr,
llvm::TargetMachine::CGFT_AssemblyFile);
llvm::CGFT_AssemblyFile);
codegen_passes.run(module);
}