Simplify GVN storage checker to use a single bitset

Author: ohadravid

compiler/rustc_mir_transform/src/gvn.rs

@@ -151,7 +151,7 @@ impl<'tcx> crate::MirPass<'tcx> for GVN {
                 .into_results_cursor(body);
 
             let mut storage_checker = StorageChecker {
-                storage_to_check: state.reused_locals.clone(),
+                reused_locals: &state.reused_locals,
                 storage_to_remove: DenseBitSet::new_empty(body.local_decls.len()),
                 maybe_uninit,
             };
@@ -1908,7 +1908,7 @@ impl<'tcx> MutVisitor<'tcx> for StorageRemover<'tcx> {
 }
 
 struct StorageChecker<'a, 'tcx> {
-    storage_to_check: DenseBitSet<Local>,
+    reused_locals: &'a DenseBitSet<Local>,
     storage_to_remove: DenseBitSet<Local>,
     maybe_uninit: ResultsCursor<'a, 'tcx, MaybeUninitializedLocals>,
 }
@@ -1928,18 +1928,16 @@ impl<'a, 'tcx> Visitor<'tcx> for StorageChecker<'a, 'tcx> {
             PlaceContext::MutatingUse(_) | PlaceContext::NonMutatingUse(_) => {}
         }
 
-        if self.storage_to_check.contains(local) {
-            self.maybe_uninit.seek_before_primary_effect(location);
-
-            if self.maybe_uninit.get().contains(local) {
-                debug!(
-                    ?location,
-                    ?local,
-                    "local is maybe uninit in this location, removing storage"
-                );
-                self.storage_to_remove.insert(local);
-                self.storage_to_check.remove(local);
-            }
+        // We only need to check reused locals which we haven't already removed storage for.
+        if !self.reused_locals.contains(local) || self.storage_to_remove.contains(local) {
+            return;
+        }
+
+        self.maybe_uninit.seek_before_primary_effect(location);
+
+        if self.maybe_uninit.get().contains(local) {
+            debug!(?location, ?local, "local is maybe uninit in this location, removing storage");
+            self.storage_to_remove.insert(local);
         }
     }
 }