cargo-pgo 0.3.0

Cargo subcommand for optimizing Rust binaries with PGO and BOLT.
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48

use crate::{resolve_binary, run_command};
use colored::Colorize;
use std::path::{Path, PathBuf};

#[derive(Debug)]
pub struct PgoEnv {
    pub llvm_profdata: PathBuf,
}

pub fn find_pgo_env() -> anyhow::Result<PgoEnv> {
    // Try to resolve `llvm-profdata` from `llvm-tools-preview`
    let path = run_command("rustc", &["--print", "target-libdir"])?
        .ok()?
        .stdout;

    let mut libpath = PathBuf::from(path);
    libpath.pop();
    libpath.push("bin/llvm-profdata");

    if libpath.exists() {
        return Ok(PgoEnv {
            llvm_profdata: libpath,
        });
    }

    libpath.set_file_name("llvm-profdata.exe");

    if libpath.exists() {
        return Ok(PgoEnv {
            llvm_profdata: libpath,
        });
    }

    // Try to find `llvm-profdata` directly in PATH
    if let Ok(llvm_profdata) = resolve_binary(Path::new("llvm-profdata"))
        .or_else(|_| resolve_binary(Path::new("llvm-profdata.exe")))
    {
        log::warn!(
            "llvm-profdata was resolved from PATH. \
Make sure that its version is compatible with rustc! If not, run `{}`.",
            "rustup component add llvm-tools-preview".blue()
        );

        Ok(PgoEnv { llvm_profdata })
    } else {
        Err(anyhow::anyhow!("Could not find `llvm-profdata`"))
    }
}