From eaf1d566ffb5e4550f8a5150accc3bf0b5210f8a Mon Sep 17 00:00:00 2001
From: Rafael Espindola <rafael.espindola@gmail.com>
Date: Thu, 3 Aug 2017 02:16:33 +0000
Subject: [PATCH] Update for llvm change.

llvm-svn: 309913
---
 lld/include/lld/Core/TargetOptionsCommandFlags.h | 3 ++-
 lld/lib/Core/TargetOptionsCommandFlags.cpp       | 4 ++--
 2 files changed, 4 insertions(+), 3 deletions(-)

diff --git a/lld/include/lld/Core/TargetOptionsCommandFlags.h b/lld/include/lld/Core/TargetOptionsCommandFlags.h
index 9ba99d94b957..9c4ff7cea3fb 100644
--- a/lld/include/lld/Core/TargetOptionsCommandFlags.h
+++ b/lld/include/lld/Core/TargetOptionsCommandFlags.h
@@ -11,10 +11,11 @@
 //
 //===----------------------------------------------------------------------===//
 
+#include "llvm/ADT/Optional.h"
 #include "llvm/Support/CodeGen.h"
 #include "llvm/Target/TargetOptions.h"
 
 namespace lld {
 llvm::TargetOptions InitTargetOptionsFromCodeGenFlags();
-llvm::CodeModel::Model GetCodeModelFromCMModel();
+llvm::Optional<llvm::CodeModel::Model> GetCodeModelFromCMModel();
 }
diff --git a/lld/lib/Core/TargetOptionsCommandFlags.cpp b/lld/lib/Core/TargetOptionsCommandFlags.cpp
index e0f26761e705..333c283fae83 100644
--- a/lld/lib/Core/TargetOptionsCommandFlags.cpp
+++ b/lld/lib/Core/TargetOptionsCommandFlags.cpp
@@ -27,6 +27,6 @@ llvm::TargetOptions lld::InitTargetOptionsFromCodeGenFlags() {
   return ::InitTargetOptionsFromCodeGenFlags();
 }
 
-llvm::CodeModel::Model lld::GetCodeModelFromCMModel() {
-  return CMModel;
+llvm::Optional<llvm::CodeModel::Model> lld::GetCodeModelFromCMModel() {
+  return getCodeModel();
 }