mirrors/ruff
1
0
Fork 0
mirror of https://github.com/astral-sh/ruff.git synced 2025-09-29 21:34:57 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 3

[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
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
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::{ use ruff_benchmark::criterion::{
criterion_group, criterion_main, BatchSize, Criterion, Throughput, 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::parsed::parsed_module;
use ruff_db::program::{ProgramSettings, SearchPathSettings, TargetVersion}; use ruff_db::program::{ProgramSettings, SearchPathSettings, TargetVersion};
use ruff_db::system::{MemoryFileSystem, SystemPath, TestSystem}; use ruff_db::system::{MemoryFileSystem, SystemPath, TestSystem};
use ruff_db::vendored::VendoredPath;
use ruff_db::Upcast; use ruff_db::Upcast;
static FOO_CODE: &str = r#" static FOO_CODE: &str = r#"
@ -48,6 +49,7 @@ struct Case {
foo: File, foo: File,
bar: File, bar: File,
typing: File, typing: File,
builtins: File,
} }
fn setup_case() -> Case { fn setup_case() -> Case {
@ -56,6 +58,7 @@ fn setup_case() -> Case {
let foo_path = SystemPath::new("/src/foo.py"); let foo_path = SystemPath::new("/src/foo.py");
let bar_path = SystemPath::new("/src/bar.py"); let bar_path = SystemPath::new("/src/bar.py");
let typing_path = SystemPath::new("/src/typing.pyi"); let typing_path = SystemPath::new("/src/typing.pyi");
let builtins_path = VendoredPath::new("stdlib/builtins.pyi");
fs.write_files([ fs.write_files([
(foo_path, FOO_CODE), (foo_path, FOO_CODE),
(bar_path, BAR_CODE), (bar_path, BAR_CODE),
@ -82,6 +85,7 @@ fn setup_case() -> Case {
let bar = system_path_to_file(&db, bar_path).unwrap(); let bar = system_path_to_file(&db, bar_path).unwrap();
let typing = system_path_to_file(&db, typing_path).unwrap(); let typing = system_path_to_file(&db, typing_path).unwrap();
let builtins = vendored_path_to_file(&db, builtins_path).unwrap();
Case { Case {
db, db,
@ -89,6 +93,7 @@ fn setup_case() -> Case {
foo, foo,
bar, bar,
typing, 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.foo);
parsed_module(case.db.upcast(), case.bar); parsed_module(case.db.upcast(), case.bar);
parsed_module(case.db.upcast(), case.typing); parsed_module(case.db.upcast(), case.typing);
parsed_module(case.db.upcast(), case.builtins);
case case
}, },
|case| { |case| {
@ -172,7 +178,7 @@ fn benchmark_cold(criterion: &mut Criterion) {
group.finish(); group.finish();
} }
criterion_group!(cold, benchmark_without_parse); criterion_group!(cold, benchmark_cold);
criterion_group!(without_parse, benchmark_cold); criterion_group!(without_parse, benchmark_without_parse);
criterion_group!(incremental, benchmark_incremental); criterion_group!(incremental, benchmark_incremental);
criterion_main!(without_parse, cold, incremental); criterion_main!(without_parse, cold, incremental);