Avoid crash with release_available_cached_blocks (pytorch#162269)

morrison-turnansky · pytorchmergebot · commit 72e6717d00ca · 2025-09-08T17:46:43.000Z
updated release behavior for cached blocks Fixes pytorch#159567 Pull Request resolved: pytorch#162269 Approved by: https://github.com/eqy, https://github.com/Skylion007
diff --git a/c10/cuda/CUDACachingAllocator.cpp b/c10/cuda/CUDACachingAllocator.cpp
@@ -3237,8 +3237,8 @@ class DeviceCachingAllocator {
           --it;
         }
         if (!(*cur)->expandable_segment_) {
-          release_block(*cur, context);
           totalReleased += (*cur)->size;
+          release_block(*cur, context);
         }
         if (is_first) {
           break;

Original file line number	Diff line number	Diff line change
`@@ -3237,8 +3237,8 @@ class DeviceCachingAllocator {`
`3237`	`3237`	`--it;`
`3238`	`3238`	`}`
`3239`	`3239`	`if (!(*cur)->expandable_segment_) {`
`3240`		`- release_block(*cur, context);`
`3241`	`3240`	`totalReleased += (*cur)->size;`
	`3241`	`+ release_block(*cur, context);`
`3242`	`3242`	`}`
`3243`	`3243`	`if (is_first) {`
`3244`	`3244`	`break;`