From 792e8f3c02a054e686e11b9d43f2dfac54f7bb75 Mon Sep 17 00:00:00 2001
From: David Blaikie <dblaikie@gmail.com>
Date: Thu, 11 Sep 2014 00:08:54 +0000
Subject: [PATCH] Use CostPool::PoolRef typedef some more

Cleanup to 217563 suggested by Lang Hames in post-commit review.

llvm-svn: 217564
---
 llvm/include/llvm/CodeGen/PBQP/CostAllocator.h | 8 +++-----
 1 file changed, 3 insertions(+), 5 deletions(-)

diff --git a/llvm/include/llvm/CodeGen/PBQP/CostAllocator.h b/llvm/include/llvm/CodeGen/PBQP/CostAllocator.h
index 67ae5ec0a1f4..02ad0d38a24c 100644
--- a/llvm/include/llvm/CodeGen/PBQP/CostAllocator.h
+++ b/llvm/include/llvm/CodeGen/PBQP/CostAllocator.h
@@ -68,19 +68,17 @@ private:
   void removeEntry(PoolEntry *p) { entrySet.erase(p); }
 
 public:
-  template <typename CostKeyT>
-  std::shared_ptr<CostT> getCost(CostKeyT costKey) {
+  template <typename CostKeyT> PoolRef getCost(CostKeyT costKey) {
     typename EntrySet::iterator itr =
       std::lower_bound(entrySet.begin(), entrySet.end(), costKey,
                        EntryComparator());
 
     if (itr != entrySet.end() && costKey == (*itr)->getCost())
-      return std::shared_ptr<CostT>((*itr)->shared_from_this(),
-                                    &(*itr)->getCost());
+      return PoolRef((*itr)->shared_from_this(), &(*itr)->getCost());
 
     auto p = std::make_shared<PoolEntry>(*this, std::move(costKey));
     entrySet.insert(itr, p.get());
-    return std::shared_ptr<CostT>(std::move(p), &p->getCost());
+    return PoolRef(std::move(p), &p->getCost());
   }
 };