Fix bad change in a CUDACachingAllocator loop (#59903)

r-barnes · facebook-github-bot · commit 25b66c64573b · 2021-06-11T17:46:04.000-07:00
Summary: Pull Request resolved: #59903 D29034650 (cf0c4ac) probably breaks something because it changes a `for` loop on ~Line 1200 from `[size,max)` to `[0,max)`. This fixes that Test Plan: Sandcastle Differential Revision: D29081688 fbshipit-source-id: 7057e3eb3737daa9d296f6a87ce2bd7de8fe997c
diff --git a/c10/cuda/CUDACachingAllocator.cpp b/c10/cuda/CUDACachingAllocator.cpp
@@ -1195,7 +1195,7 @@ class THCCachingAllocator {
     const auto size = static_cast<int64_t>(device_allocator.size());
     if (size < device_count) {
       device_allocator.resize(device_count);
-      for (const auto i : c10::irange(device_count)) {
+      for (const auto i : c10::irange(size, device_count)) {
         device_allocator[i] = std::unique_ptr<DeviceCachingAllocator>(
             new DeviceCachingAllocator());
       }
