mirror of
https://github.com/astral-sh/ruff.git
synced 2025-11-02 04:48:07 +00:00
ec863bcde7
<!--
Thank you for contributing to Ruff/ty! To help us out with reviewing,
please consider the following:
- Does this pull request include a summary of the change? (See below.)
- Does this pull request include a descriptive title? (Please prefix
with `[ty]` for ty pull
requests.)
- Does this pull request include references to any relevant issues?
-->
## Summary
<!-- What's the purpose of the change? What does it do, and why? -->
This PR adds support for building loongarch64 binaries in CI. As such
support has been merged in uv (astral-sh/uv#15387) it's time to consider
adding it to ruff.
Please note that as Ubuntu is not yet available for loongarch64, I have
elected to use a Debian Trixie container maintained by community
members. In addition, as Debian's pip does not allow installing modules
system-wide, I have modified the workflow to install additional modules
in a virtual environment.
Since the workflow is shared between all targets, the only way to handle
this difference (between Debian and Ubuntu) is just to install pip in a
venv for all targets. If there is a better (and less intrusive) way to
work around this, please let me know.
## Test Plan
Tests are included in CI and the loongarch64 artifacts built in [this
workflow](5012547154)
has been smoke tested.
68 lines
2 KiB
Rust
68 lines
2 KiB
Rust
use ruff_benchmark::criterion;
|
|
|
|
use criterion::{
|
|
BenchmarkId, Criterion, Throughput, criterion_group, criterion_main, measurement::WallTime,
|
|
};
|
|
use ruff_benchmark::{
|
|
LARGE_DATASET, NUMPY_CTYPESLIB, NUMPY_GLOBALS, PYDANTIC_TYPES, TestCase, UNICODE_PYPINYIN,
|
|
};
|
|
use ruff_python_parser::{Mode, TokenKind, lexer};
|
|
|
|
#[cfg(target_os = "windows")]
|
|
#[global_allocator]
|
|
static GLOBAL: mimalloc::MiMalloc = mimalloc::MiMalloc;
|
|
|
|
#[cfg(all(
|
|
not(target_os = "windows"),
|
|
not(target_os = "openbsd"),
|
|
any(
|
|
target_arch = "x86_64",
|
|
target_arch = "aarch64",
|
|
target_arch = "powerpc64",
|
|
target_arch = "riscv64",
|
|
target_arch = "loongarch64"
|
|
)
|
|
))]
|
|
#[global_allocator]
|
|
static GLOBAL: tikv_jemallocator::Jemalloc = tikv_jemallocator::Jemalloc;
|
|
|
|
fn create_test_cases() -> Vec<TestCase> {
|
|
vec![
|
|
TestCase::fast(NUMPY_GLOBALS.clone()),
|
|
TestCase::fast(UNICODE_PYPINYIN.clone()),
|
|
TestCase::normal(PYDANTIC_TYPES.clone()),
|
|
TestCase::normal(NUMPY_CTYPESLIB.clone()),
|
|
TestCase::slow(LARGE_DATASET.clone()),
|
|
]
|
|
}
|
|
|
|
fn benchmark_lexer(criterion: &mut Criterion<WallTime>) {
|
|
let test_cases = create_test_cases();
|
|
let mut group = criterion.benchmark_group("lexer");
|
|
|
|
for case in test_cases {
|
|
group.throughput(Throughput::Bytes(case.code().len() as u64));
|
|
group.bench_with_input(
|
|
BenchmarkId::from_parameter(case.name()),
|
|
&case,
|
|
|b, case| {
|
|
b.iter(|| {
|
|
let mut lexer = lexer::lex(case.code(), Mode::Module);
|
|
loop {
|
|
let token = lexer.next_token();
|
|
match token {
|
|
TokenKind::EndOfFile => break,
|
|
TokenKind::Unknown => panic!("Input to be a valid Python source code"),
|
|
_ => {}
|
|
}
|
|
}
|
|
});
|
|
},
|
|
);
|
|
}
|
|
|
|
group.finish();
|
|
}
|
|
|
|
criterion_group!(lexer, benchmark_lexer);
|
|
criterion_main!(lexer);
|