bench: glsl, spirv, and bincode inputs

This commit is contained in:
Dzmitry Malyshau 2022-02-15 20:01:12 -05:00
parent b05f648842
commit 5879f9ea73
2 changed files with 61 additions and 6 deletions

View File

@ -63,6 +63,7 @@ pp-rs = { version = "0.2.1", optional = true }
hexf-parse = { version = "0.2.1", optional = true }
[dev-dependencies]
bincode = "1"
criterion = { version = "0.3", features = ["html_reports"] }
diff = "0.1"
ron = "0.7"

View File

@ -1,7 +1,7 @@
use criterion::*;
use std::{fs, path::PathBuf};
use std::{fs, path::PathBuf, slice};
fn gather_inputs(folder: &str, extension: &str) -> Vec<String> {
fn gather_inputs(folder: &str, extension: &str) -> Vec<Box<[u8]>> {
let mut list = Vec::new();
let read_dir = PathBuf::from(env!("CARGO_MANIFEST_DIR"))
.join(folder)
@ -11,8 +11,8 @@ fn gather_inputs(folder: &str, extension: &str) -> Vec<String> {
match file_entry {
Ok(entry) => match entry.path().extension() {
Some(ostr) if &*ostr == extension => {
let input = fs::read_to_string(entry.path()).unwrap_or_default();
list.push(input);
let input = fs::read(entry.path()).unwrap_or_default();
list.push(input.into_boxed_slice());
}
_ => continue,
},
@ -25,17 +25,71 @@ fn gather_inputs(folder: &str, extension: &str) -> Vec<String> {
list
}
fn parse_glsl(stage: naga::ShaderStage, inputs: &[Box<[u8]>]) {
let mut parser = naga::front::glsl::Parser::default();
let options = naga::front::glsl::Options {
stage,
defines: Default::default(),
};
for input in inputs.iter() {
let string = std::str::from_utf8(input).unwrap();
parser.parse(&options, string).unwrap();
}
}
fn frontends(c: &mut Criterion) {
let mut group = c.benchmark_group("tests/in");
let mut group = c.benchmark_group("front");
#[cfg(all(feature = "serialize", feature = "deserialize"))]
group.bench_function("bin", |b| {
let inputs_wgsl = gather_inputs("tests/in", "wgsl");
let mut parser = naga::front::wgsl::Parser::new();
let inputs_bin = inputs_wgsl
.iter()
.map(|input| {
let string = std::str::from_utf8(input).unwrap();
let module = parser.parse(string).unwrap();
bincode::serialize(&module).unwrap()
})
.collect::<Vec<_>>();
b.iter(move || {
for input in inputs_bin.iter() {
bincode::deserialize::<naga::Module>(input).unwrap();
}
});
});
#[cfg(feature = "wgsl-in")]
group.bench_function("wgsl", |b| {
let inputs = gather_inputs("tests/in", "wgsl");
let mut parser = naga::front::wgsl::Parser::new();
b.iter(move || {
for input in inputs.iter() {
parser.parse(input).unwrap();
let string = std::str::from_utf8(input).unwrap();
parser.parse(string).unwrap();
}
});
});
#[cfg(feature = "spv-in")]
group.bench_function("spv", |b| {
let inputs = gather_inputs("tests/in/spv", "spv");
b.iter(move || {
let options = naga::front::spv::Options::default();
for input in inputs.iter() {
let spv =
unsafe { slice::from_raw_parts(input.as_ptr() as *const u32, input.len() / 4) };
let parser = naga::front::spv::Parser::new(spv.iter().cloned(), &options);
parser.parse().unwrap();
}
});
});
#[cfg(feature = "glsl-in")]
group.bench_function("glsl", |b| {
let vert = gather_inputs("tests/in/glsl", "vert");
b.iter(move || parse_glsl(naga::ShaderStage::Vertex, &vert));
let frag = gather_inputs("tests/in/glsl", "frag");
b.iter(move || parse_glsl(naga::ShaderStage::Vertex, &frag));
//let comp = gather_inputs("tests/in/glsl", "comp");
//b.iter(move || parse_glsl(naga::ShaderStage::Compute, &comp));
});
}
fn validation(_c: &mut Criterion) {}