use make_unique (#61272)

cyyever · facebook-github-bot · commit 00c4897c5174 · 2021-07-12T08:09:46.000-07:00
Summary: Pull Request resolved: #61272 Reviewed By: pbelevich Differential Revision: D29660354 Pulled By: ezyang fbshipit-source-id: f0aba1ea6983aec415915ed9b7dbced2e2b3b171
diff --git a/c10/cuda/CUDACachingAllocator.cpp b/c10/cuda/CUDACachingAllocator.cpp
@@ -799,8 +799,7 @@ class DeviceCachingAllocator {
     if (it == graph_pools.end()) {
       // mempool_id does not reference an existing pool. Make a new pool for
       // this capture.
-      graph_pools.emplace(std::make_pair(
-          mempool_id, std::unique_ptr<PrivatePool>(new PrivatePool)));
+      graph_pools.emplace(mempool_id, std::make_unique<PrivatePool>());
     } else {
       // mempool_id references an existing pool, which the current capture will
       // share. Check this pool is live (at least one other capture already
@@ -1339,8 +1338,7 @@ class THCCachingAllocator {
     if (size < device_count) {
       device_allocator.resize(device_count);
       for (const auto i : c10::irange(size, device_count)) {
-        device_allocator[i] = std::unique_ptr<DeviceCachingAllocator>(
-            new DeviceCachingAllocator());
+        device_allocator[i] = std::make_unique<DeviceCachingAllocator>();
       }
     }
   }
