61 lines
2.3 KiB
Rust
61 lines
2.3 KiB
Rust
use std::path::Path;
|
|
use std::process::Command;
|
|
|
|
/// Uses the `llvm-bolt` binary to instrument the artifact at the given `path` with BOLT.
|
|
/// When the instrumented artifact is executed, it will generate BOLT profiles into
|
|
/// `/tmp/prof.fdata.<pid>.fdata`.
|
|
/// Creates the instrumented artifact at `output_path`.
|
|
pub fn instrument_with_bolt(path: &Path, output_path: &Path) {
|
|
let status = Command::new("llvm-bolt")
|
|
.arg("-instrument")
|
|
.arg(&path)
|
|
// Make sure that each process will write its profiles into a separate file
|
|
.arg("--instrumentation-file-append-pid")
|
|
.arg("-o")
|
|
.arg(output_path)
|
|
.status()
|
|
.expect("Could not instrument artifact using BOLT");
|
|
|
|
if !status.success() {
|
|
panic!("Could not instrument {} with BOLT, exit code {:?}", path.display(), status.code());
|
|
}
|
|
}
|
|
|
|
/// Uses the `llvm-bolt` binary to optimize the artifact at the given `path` with BOLT,
|
|
/// using merged profiles from `profile_path`.
|
|
///
|
|
/// The recorded profiles have to be merged using the `merge-fdata` tool from LLVM and the merged
|
|
/// profile path should be then passed to this function.
|
|
///
|
|
/// Creates the optimized artifact at `output_path`.
|
|
pub fn optimize_with_bolt(path: &Path, profile_path: &Path, output_path: &Path) {
|
|
let status = Command::new("llvm-bolt")
|
|
.arg(&path)
|
|
.arg("-data")
|
|
.arg(&profile_path)
|
|
.arg("-o")
|
|
.arg(output_path)
|
|
// Reorder basic blocks within functions
|
|
.arg("-reorder-blocks=ext-tsp")
|
|
// Reorder functions within the binary
|
|
.arg("-reorder-functions=hfsort+")
|
|
// Split function code into hot and code regions
|
|
.arg("-split-functions")
|
|
// Split as many basic blocks as possible
|
|
.arg("-split-all-cold")
|
|
// Move jump tables to a separate section
|
|
.arg("-jump-tables=move")
|
|
// Fold functions with identical code
|
|
.arg("-icf=1")
|
|
// Update DWARF debug info in the final binary
|
|
.arg("-update-debug-sections")
|
|
// Print optimization statistics
|
|
.arg("-dyno-stats")
|
|
.status()
|
|
.expect("Could not optimize artifact using BOLT");
|
|
|
|
if !status.success() {
|
|
panic!("Could not optimize {} with BOLT, exit code {:?}", path.display(), status.code());
|
|
}
|
|
}
|