Update for LLVM api change.

llvm-svn: 216584

2 files changed, 6 insertions, 4 deletions

diff --git a/clang-tools-extra/clang-apply-replacements/tool/ClangApplyReplacementsMain.cpp b/clang-tools-extra/clang-apply-replacements/tool/ClangApplyReplacementsMain.cpp
index 4559a26a12b..58bdb035c52 100644
--- a/clang-tools-extra/clang-apply-replacements/tool/ClangApplyReplacementsMain.cpp
+++ b/clang-tools-extra/clang-apply-replacements/tool/ClangApplyReplacementsMain.cpp
@@ -188,8 +188,9 @@ bool applyFormatting(const std::vector<tooling::Replacement> &Replacements,
 
   FileManager Files((FileSystemOptions()));
   SourceManager SM(Diagnostics, Files);
-  SM.overrideFileContents(Files.getFile(FileName),
-                          llvm::MemoryBuffer::getMemBufferCopy(FileData));
+  SM.overrideFileContents(
+      Files.getFile(FileName),
+      llvm::MemoryBuffer::getMemBufferCopy(FileData).release());
   Rewriter Rewrites(SM, LangOptions());
 
   return getRewrittenData(FormattingReplacements, Rewrites, FormattedFileData);
diff --git a/clang-tools-extra/unittests/include/common/VirtualFileHelper.h b/clang-tools-extra/unittests/include/common/VirtualFileHelper.h
index 4ec00da9ced..f6fa7fcbe53 100644
--- a/clang-tools-extra/unittests/include/common/VirtualFileHelper.h
+++ b/clang-tools-extra/unittests/include/common/VirtualFileHelper.h
@@ -59,10 +59,11 @@ public:
              I = VirtualFiles.begin(),
              E = VirtualFiles.end();
          I != E; ++I) {
-      llvm::MemoryBuffer *Buf = llvm::MemoryBuffer::getMemBuffer(I->Code);
+      std::unique_ptr<llvm::MemoryBuffer> Buf =
+          llvm::MemoryBuffer::getMemBuffer(I->Code);
       const FileEntry *Entry = SM.getFileManager().getVirtualFile(
           I->FileName, Buf->getBufferSize(), /*ModificationTime=*/0);
-      SM.overrideFileContents(Entry, Buf);
+      SM.overrideFileContents(Entry, Buf.release());
     }
   }