mirror of https://github.com/rust-lang/rust
42 lines
1.5 KiB
Rust
42 lines
1.5 KiB
Rust
// This test checks the -Zremark-dir flag, which writes LLVM
|
|
// optimization remarks to the YAML format. When using PGO (Profile
|
|
// Guided Optimization), the Hotness attribute should be included in
|
|
// the output remark files.
|
|
// See https://github.com/rust-lang/rust/pull/114439
|
|
|
|
//@ needs-profiler-support
|
|
//@ ignore-cross-compile
|
|
|
|
use run_make_support::{
|
|
has_extension, has_prefix, invalid_utf8_contains, llvm_profdata, run, rustc, shallow_find_files,
|
|
};
|
|
|
|
fn main() {
|
|
rustc().profile_generate("profdata").opt().input("foo.rs").output("foo").run();
|
|
run("foo");
|
|
// The profdata filename is a long sequence of numbers, fetch it by prefix and extension
|
|
// to keep the test working even if the filename changes.
|
|
let profdata_files = shallow_find_files("profdata", |path| {
|
|
has_prefix(path, "default") && has_extension(path, "profraw")
|
|
});
|
|
let profdata_file = profdata_files.get(0).unwrap();
|
|
llvm_profdata().merge().output("merged.profdata").input(profdata_file).run();
|
|
rustc()
|
|
.profile_use("merged.profdata")
|
|
.opt()
|
|
.input("foo.rs")
|
|
.arg("-Cremark=all")
|
|
.arg("-Zremark-dir=profiles")
|
|
.run();
|
|
// Check that PGO hotness is included in the remark files
|
|
let remark_files = shallow_find_files("profiles", |path| {
|
|
has_prefix(path, "foo") && has_extension(path, "yaml")
|
|
});
|
|
assert!(!remark_files.is_empty());
|
|
for file in remark_files {
|
|
if !file.to_str().unwrap().contains("codegen") {
|
|
invalid_utf8_contains(file, "Hotness")
|
|
};
|
|
}
|
|
}
|