use ccx

tesuji · tesuji · commit dca720706d60 · 2024-06-10T06:52:34.000Z
diff --git a/compiler/rustc_mir_transform/src/promote_consts_local_arrays.rs b/compiler/rustc_mir_transform/src/promote_consts_local_arrays.rs
@@ -60,6 +60,7 @@ impl<'tcx> MirPass<'tcx> for PromoteArraysOpt<'tcx> {
 
     #[instrument(level = "trace", skip(self, tcx, body))]
     fn run_pass(&self, tcx: TyCtxt<'tcx>, body: &mut Body<'tcx>) {
+        use hir::ConstContext::{Const, Static};
         debug!("running on body: {:?}", body.source.def_id());
         // There's not really any point in promoting errorful MIR.
         //
@@ -72,16 +73,12 @@ impl<'tcx> MirPass<'tcx> for PromoteArraysOpt<'tcx> {
             return;
         }
 
+        let ccx = ConstCx::new(tcx, body);
         // Ignore static/const items. Already processed by PromoteTemps
-        if let Some(ctx) = tcx.hir().body_const_context(body.source.def_id()) {
-            use hir::ConstContext::*;
-            match ctx {
-                Static(_) | Const { inline: _ } => return,
-                _ => {}
-            }
+        if let Some(Static(_) | Const { inline: _ }) = ccx.const_kind {
+            return;
         }
 
-        let ccx = ConstCx::new(tcx, body);
         let (mut temps, mut all_candidates, already_promoted) = collect_temps_and_candidates(&ccx);
 
         let promotable_candidates = validate_candidates(&ccx, &mut temps, &mut all_candidates);
diff --git a/tests/mir-opt/const_array_locals.rs b/tests/mir-opt/const_array_locals.rs
@@ -5,7 +5,7 @@
 struct F32x8([f32; 8]);
 
 // EMIT_MIR const_array_locals.main.PromoteArraysOpt.diff
-// CHECK-LABEL: fn main() -> () {
+// CHECK-LABEL: fn main(
 #[rustfmt::skip]
 pub fn main() {
     // CHECK: let [[array_lit:_.*]]: [i32; 32];
