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

1 files changed, 6 insertions, 5 deletions

diff --git a/polly/lib/CodeGen/PTXGenerator.cpp b/polly/lib/CodeGen/PTXGenerator.cpp
index e2432b5515f9..dc1339a3a2a4 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;