compiler/rustc_codegen_cranelift/src/driver/mod.rs

   1 //! Drivers are responsible for calling [`codegen_fn`] or [`codegen_static`] for each mono item and
   2 //! performing any further actions like JIT executing or writing object files.
   3 //!
   4 //! [`codegen_fn`]: crate::base::codegen_fn
   5 //! [`codegen_static`]: crate::constant::codegen_static
   6
   7 use rustc_middle::mir::mono::{Linkage as RLinkage, MonoItem, Visibility};
   8
   9 use crate::prelude::*;
  10
  11 pub(crate) mod aot;
  12 #[cfg(feature = "jit")]
  13 pub(crate) mod jit;
  14
  15 fn predefine_mono_items<'tcx>(
  16     tcx: TyCtxt<'tcx>,
  17     module: &mut dyn Module,
  18     mono_items: &[(MonoItem<'tcx>, (RLinkage, Visibility))],
  19 ) {
  20     tcx.sess.time("predefine functions", || {
  21         let is_compiler_builtins = tcx.is_compiler_builtins(LOCAL_CRATE);
  22         for &(mono_item, (linkage, visibility)) in mono_items {
  23             match mono_item {
  24                 MonoItem::Fn(instance) => {
  25                     let name = tcx.symbol_name(instance).name;
  26                     let _inst_guard = crate::PrintOnPanic(|| format!("{:?} {}", instance, name));
  27                     let sig = get_function_sig(tcx, module.isa().triple(), instance);
  28                     let linkage = crate::linkage::get_clif_linkage(
  29                         mono_item,
  30                         linkage,
  31                         visibility,
  32                         is_compiler_builtins,
  33                     );
  34                     module.declare_function(name, linkage, &sig).unwrap();
  35                 }
  36                 MonoItem::Static(_) | MonoItem::GlobalAsm(_) => {}
  37             }
  38         }
  39     });
  40 }
  41
  42 fn time<R>(tcx: TyCtxt<'_>, display: bool, name: &'static str, f: impl FnOnce() -> R) -> R {
  43     if display {
  44         println!("[{:<30}: {}] start", tcx.crate_name(LOCAL_CRATE), name);
  45         let before = std::time::Instant::now();
  46         let res = tcx.sess.time(name, f);
  47         let after = std::time::Instant::now();
  48         println!("[{:<30}: {}] end time: {:?}", tcx.crate_name(LOCAL_CRATE), name, after - before);
  49         res
  50     } else {
  51         tcx.sess.time(name, f)
  52     }
  53 }