#llvm #marker #generate #macro #comments #proc-macro #procedural

nightly macro llvm-mca-macros

Procedural macros for generating llvm-mca marker comments

1 unstable release

0.1.0 Nov 7, 2024

#401 in Profiling

Download history 90/week @ 2024-11-02 25/week @ 2024-11-09

115 downloads per month

MIT/Apache

8KB
73 lines

llvm-mca-macros

Procedural macros to generate marker comments for LLVM's Machine Code Analyzer.

These macros generate markers after the function epilogue and before the function prologue. If more granularity is needed, you can use the llvm-mca crate instead.

Usage

By default, llvm_mca will disable inlining. For example, this:

use llvm_mca_macros::llvm_mca;

#[llvm_mca]
fn quadruple(x: u32) -> u32 {
    let doubled = x + x;
    doubled + doubled
}

will generate the equivalent of:

#[inline(never)]
fn quadruple(x: u32) -> u32 {
    // emit `LLVM-MCA-BEGIN` marker
    let ret = {
        let doubled = x + x;
        doubled + doubled
    };
    // emit `LLVM-MCA-END` marker
    ret
}

If inlining is desired, the allow_inline attribute can be specified:

use llvm_mca_macros::llvm_mca;

#[llvm_mca(allow_inline)]
fn quadruple(x: u32) -> u32 {
    let doubled = x + x;
    doubled + doubled
}

This will generate the equivalent of:

fn quadruple(x: u32) -> u32 {
    // emit `LLVM-MCA-BEGIN` marker
    let ret = {
        let doubled = x + x;
        doubled + doubled
    };
    // emit `LLVM-MCA-END` marker
    ret
}

Generating assembly

You must set the RUSTFLAGS="--emit asm" option when building your project with cargo. For example:

RUSTFLAGS="--emit asm" cargo build --release

This will output assembly files in target/*/deps

Dependencies

~225–670KB
~16K SLoC