mirror of
https://github.com/italicsjenga/vello.git
synced 2025-01-10 20:51:29 +11:00
c6965de557
Measure bandwidth of building linked lists with atomics.
128 lines
3.9 KiB
Rust
128 lines
3.9 KiB
Rust
// Copyright 2021 The piet-gpu authors.
|
|
//
|
|
// Licensed under the Apache License, Version 2.0 (the "License");
|
|
// you may not use this file except in compliance with the License.
|
|
// You may obtain a copy of the License at
|
|
//
|
|
// https://www.apache.org/licenses/LICENSE-2.0
|
|
//
|
|
// Unless required by applicable law or agreed to in writing, software
|
|
// distributed under the License is distributed on an "AS IS" BASIS,
|
|
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
// See the License for the specific language governing permissions and
|
|
// limitations under the License.
|
|
//
|
|
// Also licensed under MIT license, at your choice.
|
|
|
|
//! Tests for piet-gpu shaders and GPU capabilities.
|
|
|
|
mod clear;
|
|
mod config;
|
|
mod linkedlist;
|
|
mod message_passing;
|
|
mod prefix;
|
|
mod prefix_tree;
|
|
mod runner;
|
|
mod test_result;
|
|
|
|
use clap::{App, Arg};
|
|
use piet_gpu_hal::InstanceFlags;
|
|
|
|
use crate::config::Config;
|
|
pub use crate::runner::Runner;
|
|
use crate::test_result::ReportStyle;
|
|
pub use crate::test_result::TestResult;
|
|
|
|
fn main() {
|
|
let matches = App::new("piet-gpu-tests")
|
|
.arg(
|
|
Arg::with_name("verbose")
|
|
.short("v")
|
|
.long("verbose")
|
|
.help("Verbose reporting of results"),
|
|
)
|
|
.arg(
|
|
Arg::with_name("groups")
|
|
.short("g")
|
|
.long("groups")
|
|
.help("Groups to run")
|
|
.takes_value(true),
|
|
)
|
|
.arg(
|
|
Arg::with_name("size")
|
|
.short("s")
|
|
.long("size")
|
|
.help("Size of tests")
|
|
.takes_value(true),
|
|
)
|
|
.arg(
|
|
Arg::with_name("n_iter")
|
|
.short("n")
|
|
.long("n_iter")
|
|
.help("Number of iterations")
|
|
.takes_value(true),
|
|
)
|
|
.arg(
|
|
Arg::with_name("dx12")
|
|
.long("dx12")
|
|
.help("Prefer DX12 backend"),
|
|
)
|
|
.get_matches();
|
|
let style = if matches.is_present("verbose") {
|
|
ReportStyle::Verbose
|
|
} else {
|
|
ReportStyle::Short
|
|
};
|
|
let config = Config::from_matches(&matches);
|
|
unsafe {
|
|
let report = |test_result: &TestResult| {
|
|
test_result.report(style);
|
|
};
|
|
let mut flags = InstanceFlags::empty();
|
|
if matches.is_present("dx12") {
|
|
flags |= InstanceFlags::DX12;
|
|
}
|
|
let mut runner = Runner::new(flags);
|
|
if style == ReportStyle::Verbose {
|
|
// TODO: get adapter name in here too
|
|
println!("Backend: {:?}", runner.backend_type());
|
|
}
|
|
report(&clear::run_clear_test(&mut runner, &config));
|
|
if config.groups.matches("prefix") {
|
|
report(&prefix::run_prefix_test(
|
|
&mut runner,
|
|
&config,
|
|
prefix::Variant::Compatibility,
|
|
));
|
|
report(&prefix::run_prefix_test(
|
|
&mut runner,
|
|
&config,
|
|
prefix::Variant::Atomic,
|
|
));
|
|
if runner.session.gpu_info().has_memory_model {
|
|
report(&prefix::run_prefix_test(
|
|
&mut runner,
|
|
&config,
|
|
prefix::Variant::Vkmm,
|
|
));
|
|
}
|
|
report(&prefix_tree::run_prefix_test(&mut runner, &config));
|
|
}
|
|
if config.groups.matches("atomic") {
|
|
report(&message_passing::run_message_passing_test(
|
|
&mut runner,
|
|
&config,
|
|
message_passing::Variant::Atomic,
|
|
));
|
|
if runner.session.gpu_info().has_memory_model {
|
|
report(&message_passing::run_message_passing_test(
|
|
&mut runner,
|
|
&config,
|
|
message_passing::Variant::Vkmm,
|
|
));
|
|
}
|
|
report(&linkedlist::run_linkedlist_test(&mut runner, &config));
|
|
}
|
|
}
|
|
}
|