nordic-dev.net/rust
nordic-dev.net/rust
2
0
Fork 0
mirror of https://github.com/rust-lang/rust.git synced 2025-04-15 05:26:47 +00:00
Code Issues Packages Projects Releases Wiki Activity

Add explicit inlining to methods after AD layer

Browse Source
This commit is contained in:
bit-aloo 2025-04-12 17:23:19 +05:30
parent 0af84744f8
commit 530b56523b
No known key found for this signature in database
GPG Key ID: 02911B24FDAE81DA
1 changed files with 16 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

18
compiler/rustc_codegen_llvm/src/back/lto.rs
View File

@ -28,8 +28,9 @@ use crate::back::write::{
use crate::errors::{
DynamicLinkingWithLTO, LlvmError, LtoBitcodeFromRlib, LtoDisallowed, LtoDylib, LtoProcMacro,
};
use crate::llvm::{self, build_string};
use crate::{LlvmCodegenBackend, ModuleLlvm};
use crate::llvm::AttributePlace::Function;
use crate::llvm::{self, build_string, get_value_name};
use crate::{LlvmCodegenBackend, ModuleLlvm, SimpleCx, attributes};
/// We keep track of the computed LTO cache keys from the previous
/// session to determine which CGUs we can reuse.
@ -664,6 +665,19 @@ pub(crate) fn run_pass_manager(
unsafe { llvm::LLVMDumpModule(module.module_llvm.llmod()) };
}
let cx =
SimpleCx::new(module.module_llvm.llmod(), &module.module_llvm.llcx, cgcx.pointer_size);
for function in cx.get_functions() {
let name = get_value_name(function);
let name = std::str::from_utf8(name).unwrap();
if name.starts_with("__enzyme") {
let attr = llvm::AttributeKind::AlwaysInline.create_attr(cx.llcx);
attributes::apply_to_llfn(function, Function, &[attr]);
}
}
let opt_stage = llvm::OptStage::FatLTO;
let stage = write::AutodiffStage::PostAD;
unsafe {