vyre-driver-wgpu 0.7.1

wgpu backend for vyre IR - implements VyreBackend, owns GPU runtime, buffer pool, pipeline cache
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
39
40
41
42
43
44
45
46
47

//! Hardware WGPU parity tests for C comment-strip masking.

#![allow(deprecated)]
use vyre_driver::{DispatchConfig, VyreBackend};
use vyre_libs::parsing::c::preprocess::gpu_comment_strip_mask::{
    gpu_comment_strip_mask, reference_gpu_comment_strip_mask,
};

fn pack_source_bytes(source: &[u8]) -> Vec<u8> {
    let mut packed = source.to_vec();
    packed.resize((source.len().max(1).div_ceil(4) * 4).max(4), 0);
    packed
}

fn unpack_u32_prefix(bytes: &[u8], count: usize) -> Vec<u32> {
    bytes
        .chunks_exact(4)
        .take(count)
        .map(|chunk| u32::from_le_bytes([chunk[0], chunk[1], chunk[2], chunk[3]]))
        .collect()
}

fn run_gpu_comment_mask(source: &[u8]) -> Vec<u32> {
    let program = gpu_comment_strip_mask(source.len() as u32);
    let inputs = vec![
        pack_source_bytes(source),
        vec![0u8; source.len().max(1) * 4],
    ];
    let backend = vyre_driver_wgpu::WgpuBackend::acquire()
        .expect("Fix: WGPU backend must acquire the local GPU for C comment-strip parity tests");
    let outputs = backend
        .dispatch(&program, &inputs, &DispatchConfig::default())
        .expect("Fix: gpu_comment_strip_mask must dispatch on the WGPU backend");
    let bytes = outputs
        .first()
        .expect("Fix: gpu_comment_strip_mask must return comment_mask_out");
    unpack_u32_prefix(bytes, source.len())
}

#[test]
fn wgpu_comment_strip_handles_realistic_c_snippet_on_device() {
    let source = b"// header guard\n#ifndef X\n#define X /* opaque */\nchar *s = \"/* not comment */\";\nint c = '/';\n#endif\n";
    assert_eq!(
        run_gpu_comment_mask(source),
        reference_gpu_comment_strip_mask(source)
    );
}