Use CostPool::PoolRef typedef some more

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

git-svn-id: https://llvm.org/svn/llvm-project/llvm/trunk@217564 91177308-0d34-0410-b5e6-96231b3b80d8

diff --git a/include/llvm/CodeGen/PBQP/CostAllocator.h b/include/llvm/CodeGen/PBQP/CostAllocator.h
index 67ae5ec0a1f4ac0a713065f8baedb05ac9bc44db..02ad0d38a24c218da895ef9615a945ac7944b796 100644 (file)
--- a/include/llvm/CodeGen/PBQP/CostAllocator.h
+++ b/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());
   }
 };