wgpu-primitives 0.3.0

Composable GPU primitives for Rust applications using wgpu.
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38

use criterion::{BenchmarkGroup, Throughput, measurement::WallTime};
use rand::{Rng, SeedableRng, rngs::StdRng};
use wgpu_primitives::Context;

pub const INPUT_SIZES: [usize; 4] = [100_000, 1_000_000, 10_000_000, 100_000_000];

pub fn seeded_input(item_count: usize, seed: u64) -> Vec<u32> {
    let mut rng = StdRng::seed_from_u64(seed ^ item_count as u64);
    (0..item_count).map(|_| rng.random()).collect()
}

pub fn report_adapter(context: &Context) {
    let info = &context.adapter_info;
    eprintln!(
        "wgpu adapter: name={:?} backend={:?} device_type={:?} driver={:?} driver_info={:?}",
        info.name, info.backend, info.device_type, info.driver, info.driver_info
    );
}

pub fn configure_group(group: &mut BenchmarkGroup<'_, WallTime>, item_count: usize) {
    group.throughput(Throughput::Elements(item_count as u64));
    if item_count >= 10_000_000 {
        group.sample_size(10);
        group.measurement_time(std::time::Duration::from_secs(15));
    } else {
        group.sample_size(30);
        group.measurement_time(std::time::Duration::from_secs(5));
    }
}

pub fn wait_for_gpu(device: &wgpu::Device) {
    device
        .poll(wgpu::PollType::Wait {
            submission_index: None,
            timeout: None,
        })
        .expect("failed to wait for GPU work");
}