vyre-conform 0.1.0

Conformance suite for vyre backends — proves byte-identical output to CPU reference
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

//! Synchronous readback helper for wgpu buffer maps.
//!
//! wgpu buffer mapping is asynchronous, but conformance tests are written
//! synchronously. [`wait_for_readback`] polls the device until the map callback
//! fires or a timeout expires, yielding a clear actionable error on either
//! device loss or a wedged shader.

use std::time::{Duration, Instant};

const DISPATCH_TIMEOUT: Duration = Duration::from_secs(30);

pub(super) fn wait_for_readback(
    device: &wgpu::Device,
    receiver: std::sync::mpsc::Receiver<Result<(), wgpu::BufferAsyncError>>,
) -> Result<Result<(), wgpu::BufferAsyncError>, String> {
    let deadline = Instant::now() + DISPATCH_TIMEOUT;
    loop {
        device.poll(wgpu::Maintain::Poll);
        match receiver.try_recv() {
            Ok(result) => return Ok(result),
            Err(std::sync::mpsc::TryRecvError::Disconnected) => {
                return Err(
                    "Fix: GPU readback channel closed before map_async completed. Keep the readback receiver alive until the callback fires."
                        .to_string(),
                );
            }
            Err(std::sync::mpsc::TryRecvError::Empty) => {}
        }
        if Instant::now() >= deadline {
            return Err(format!(
                "Fix: GPU dispatch/readback timed out after {}s. Check for device loss, a wedged adapter, or a shader that never completes.",
                DISPATCH_TIMEOUT.as_secs()
            ));
        }
        std::thread::sleep(Duration::from_millis(1));
    }
}