Cyclic0007/ruff
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

[red-knot] preparse builtins in without_parse benchmark (#12395)

Browse Source
This commit is contained in:
Carl Meyer
2024-07-18 22:58:27 -07:00
committed by GitHub
parent d61747093c
commit a62e2d2000
1 changed files with 9 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
crates/ruff_benchmark/benches/red_knot.rs
View File

@@ -5,10 +5,11 @@ use red_knot::workspace::WorkspaceMetadata;
use ruff_benchmark::criterion::{
criterion_group, criterion_main, BatchSize, Criterion, Throughput,
};
use ruff_db::files::{system_path_to_file, File};
use ruff_db::files::{system_path_to_file, vendored_path_to_file, File};
use ruff_db::parsed::parsed_module;
use ruff_db::program::{ProgramSettings, SearchPathSettings, TargetVersion};
use ruff_db::system::{MemoryFileSystem, SystemPath, TestSystem};
use ruff_db::vendored::VendoredPath;
use ruff_db::Upcast;
static FOO_CODE: &str = r#"
@@ -48,6 +49,7 @@ struct Case {
foo: File,
bar: File,
typing: File,
builtins: File,
}
fn setup_case() -> Case {
@@ -56,6 +58,7 @@ fn setup_case() -> Case {
let foo_path = SystemPath::new("/src/foo.py");
let bar_path = SystemPath::new("/src/bar.py");
let typing_path = SystemPath::new("/src/typing.pyi");
let builtins_path = VendoredPath::new("stdlib/builtins.pyi");
fs.write_files([
(foo_path, FOO_CODE),
(bar_path, BAR_CODE),
@@ -82,6 +85,7 @@ fn setup_case() -> Case {
let bar = system_path_to_file(&db, bar_path).unwrap();
let typing = system_path_to_file(&db, typing_path).unwrap();
let builtins = vendored_path_to_file(&db, builtins_path).unwrap();
Case {
db,
@@ -89,6 +93,7 @@ fn setup_case() -> Case {
foo,
bar,
typing,
builtins,
}
}
@@ -104,6 +109,7 @@ fn benchmark_without_parse(criterion: &mut Criterion) {
parsed_module(case.db.upcast(), case.foo);
parsed_module(case.db.upcast(), case.bar);
parsed_module(case.db.upcast(), case.typing);
parsed_module(case.db.upcast(), case.builtins);
case
},
|case| {
@@ -172,7 +178,7 @@ fn benchmark_cold(criterion: &mut Criterion) {
group.finish();
}
criterion_group!(cold, benchmark_without_parse);
criterion_group!(without_parse, benchmark_cold);
criterion_group!(cold, benchmark_cold);
criterion_group!(without_parse, benchmark_without_parse);
criterion_group!(incremental, benchmark_incremental);
criterion_main!(without_parse, cold, incremental);