nordic-dev.net/wgpu
nordic-dev.net/wgpu
2
0
Fork 0
mirror of https://github.com/gfx-rs/wgpu.git synced 2024-11-29 10:13:34 +00:00
Code Issues Packages Projects Releases Wiki Activity

Isolate hello-compute examples to own file

Browse Source
This commit is contained in:
Connor Fitzgerald 2021-06-21 15:39:33 -04:00
parent f47a54009d
commit ccef63e914
2 changed files with 92 additions and 93 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

94
wgpu/examples/hello-compute/main.rs
View File

@ -196,96 +196,4 @@ fn main() {
}
#[cfg(all(test, not(target_arch = "wasm32")))]
mod tests {
#[path = "../../../tests/common/mod.rs"]
mod common;
use std::sync::Arc;
use super::*;
use common::{initialize_test, TestParameters};
#[test]
fn test_compute_1() {
initialize_test(TestParameters::default(), |ctx| {
let input = &[1, 2, 3, 4];
pollster::block_on(assert_execute_gpu(
&ctx.device,
&ctx.queue,
input,
&[0, 1, 7, 2],
));
});
}
#[test]
fn test_compute_2() {
initialize_test(TestParameters::default(), |ctx| {
let input = &[5, 23, 10, 9];
pollster::block_on(assert_execute_gpu(
&ctx.device,
&ctx.queue,
input,
&[5, 15, 6, 19],
));
});
}
#[test]
fn test_compute_overflow() {
initialize_test(TestParameters::default(), |ctx| {
let input = &[77031, 837799, 8400511, 63728127];
pollster::block_on(assert_execute_gpu(
&ctx.device,
&ctx.queue,
input,
&[350, 524, OVERFLOW, OVERFLOW],
));
});
}
#[test]
fn test_multithreaded_compute() {
initialize_test(TestParameters::default(), |ctx| {
use std::{sync::mpsc, thread, time::Duration};
let ctx = Arc::new(ctx);
let thread_count = 8;
let (tx, rx) = mpsc::channel();
for _ in 0..thread_count {
let tx = tx.clone();
let ctx = Arc::clone(&ctx);
thread::spawn(move || {
let input = &[100, 100, 100];
pollster::block_on(assert_execute_gpu(
&ctx.device,
&ctx.queue,
input,
&[25, 25, 25],
));
tx.send(true).unwrap();
});
}
for _ in 0..thread_count {
rx.recv_timeout(Duration::from_secs(10))
.expect("A thread never completed.");
}
});
}
async fn assert_execute_gpu(
device: &wgpu::Device,
queue: &wgpu::Queue,
input: &[u32],
expected: &[u32],
) {
if let Some(produced) = execute_gpu_inner(device, queue, input).await {
assert_eq!(produced, expected);
}
}
}
mod tests;

91
wgpu/examples/hello-compute/tests.rs Normal file
View File

@ -0,0 +1,91 @@
#[path = "../../tests/common/mod.rs"]
mod common;
use std::sync::Arc;
use super::*;
use common::{initialize_test, TestParameters};
#[test]
fn test_compute_1() {
initialize_test(TestParameters::default(), |ctx| {
let input = &[1, 2, 3, 4];
pollster::block_on(assert_execute_gpu(
&ctx.device,
&ctx.queue,
input,
&[0, 1, 7, 2],
));
});
}
#[test]
fn test_compute_2() {
initialize_test(TestParameters::default(), |ctx| {
let input = &[5, 23, 10, 9];
pollster::block_on(assert_execute_gpu(
&ctx.device,
&ctx.queue,
input,
&[5, 15, 6, 19],
));
});
}
#[test]
fn test_compute_overflow() {
initialize_test(TestParameters::default(), |ctx| {
let input = &[77031, 837799, 8400511, 63728127];
pollster::block_on(assert_execute_gpu(
&ctx.device,
&ctx.queue,
input,
&[350, 524, OVERFLOW, OVERFLOW],
));
});
}
#[test]
fn test_multithreaded_compute() {
initialize_test(TestParameters::default(), |ctx| {
use std::{sync::mpsc, thread, time::Duration};
let ctx = Arc::new(ctx);
let thread_count = 8;
let (tx, rx) = mpsc::channel();
for _ in 0..thread_count {
let tx = tx.clone();
let ctx = Arc::clone(&ctx);
thread::spawn(move || {
let input = &[100, 100, 100];
pollster::block_on(assert_execute_gpu(
&ctx.device,
&ctx.queue,
input,
&[25, 25, 25],
));
tx.send(true).unwrap();
});
}
for _ in 0..thread_count {
rx.recv_timeout(Duration::from_secs(10))
.expect("A thread never completed.");
}
});
}
async fn assert_execute_gpu(
device: &wgpu::Device,
queue: &wgpu::Queue,
input: &[u32],
expected: &[u32],
) {
if let Some(produced) = execute_gpu_inner(device, queue, input).await {
assert_eq!(produced, expected);
}
}