print stats

Author: csmoe

compiler/rustc_codegen_llvm/src/back/write.rs

@@ -568,6 +568,8 @@ pub(crate) unsafe fn codegen(
         let llcx = &*module.module_llvm.llcx;
         let tm = &*module.module_llvm.tm;
         let module_name = module.name.clone();
+        let count = unsafe { llvm::LLVMRustModuleInstructionStats(&llmod) };
+        println!("llvm-module: {module_name} = {count}");
         let module_name = Some(&module_name[..]);
         let handlers = DiagnosticHandlers::new(cgcx, diag_handler, llcx);
 

compiler/rustc_codegen_llvm/src/llvm/ffi.rs

@@ -2410,6 +2410,7 @@ extern "C" {
     pub fn LLVMRustModuleBufferLen(p: &ModuleBuffer) -> usize;
     pub fn LLVMRustModuleBufferFree(p: &'static mut ModuleBuffer);
     pub fn LLVMRustModuleCost(M: &Module) -> u64;
+    pub fn LLVMRustModuleInstructionStats(M: &Module) -> u64;
 
     pub fn LLVMRustThinLTOBufferCreate(M: &Module, is_thin: bool) -> &'static mut ThinLTOBuffer;
     pub fn LLVMRustThinLTOBufferFree(M: &'static mut ThinLTOBuffer);

compiler/rustc_llvm/llvm-wrapper/RustWrapper.cpp

@@ -1754,10 +1754,13 @@ LLVMRustModuleCost(LLVMModuleRef M) {
 extern "C" uint64_t
 LLVMRustModuleInstructionStats(LLVMModuleRef M) {
   auto f = unwrap(M)->functions();
+  raw_fd_ostream OS(2, false); // stderr.
   for (auto &func : f)
   {
     auto name = func.getName();
+
     auto count = func.getInstructionCount();
+    OS << name << "\t" << count << "\n";
   }
   return unwrap(M)->getInstructionCount();
 }