From: Tobias Grosser <tobias@grosser.es>
Date: Mon, 27 Apr 2015 12:06:32 +0000 (+0000)
Subject: Fix PTXGenerator after raw_pwrite_stream has been introduced
X-Git-Tag: llvmorg-3.7.0-rc1~5942
X-Git-Url: http://review.tizen.org/git/?a=commitdiff_plain;h=4245372bee0af5f42296812562c64b144ca61fc2;p=platform%2Fupstream%2Fllvm.git

Fix PTXGenerator after raw_pwrite_stream has been introduced

Without this patch Polly with GPGPU support enabled did not compile any more.

llvm-svn: 235868
---

diff --git a/polly/lib/CodeGen/PTXGenerator.cpp b/polly/lib/CodeGen/PTXGenerator.cpp
index e2432b5..dc1339a 100644
--- a/polly/lib/CodeGen/PTXGenerator.cpp
+++ b/polly/lib/CodeGen/PTXGenerator.cpp
@@ -19,6 +19,7 @@
 
 #include "llvm/IR/LegacyPassManager.h"
 #include "llvm/ADT/SetVector.h"
+#include "llvm/ADT/SmallString.h"
 #include "llvm/Analysis/TargetLibraryInfo.h"
 #include "llvm/Analysis/TargetTransformInfo.h"
 #include "llvm/Support/Debug.h"
@@ -566,19 +567,19 @@ static bool createASMAsString(Module *New, const StringRef &Triple,
   PM.add(createTargetTransformInfoWrapperPass(Target.getTargetIRAnalysis()));
 
   {
-    raw_string_ostream NameROS(ASM);
-    formatted_raw_ostream FOS(NameROS);
+    SmallString<100> ASMSmall;
+    raw_svector_ostream NameROSSmall(ASMSmall);
 
     // Ask the target to add backend passes as necessary.
     int UseVerifier = true;
-    if (Target.addPassesToEmitFile(PM, FOS, TargetMachine::CGFT_AssemblyFile,
-                                   UseVerifier)) {
+    if (Target.addPassesToEmitFile(
+            PM, NameROSSmall, TargetMachine::CGFT_AssemblyFile, UseVerifier)) {
       errs() << "The target does not support generation of this file type!\n";
       return false;
     }
 
+    ASM = ASMSmall.c_str();
     PM.run(*New);
-    FOS.flush();
   }
 
   return true;