vyre 0.4.0

GPU compute intermediate representation with a standard operation library
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

//! WGSL lowering source for `workgroup.union_find`.
//!
//! This module contains the complete Rust-side lowering handle for the
//! disjoint-set forest backed by atomic parent slots. The shader body remains in the adjacent `.wgsl` asset
//! so backend loaders and tests share the exact same source bytes.
//! Keeping entry-point metadata here removes the old generated
//! `lowering.rs` plus `lowering/wgsl.rs` split.

/// Registered WGSL compute entry point for this intrinsic.
pub const ENTRY_POINT: &str = "workgroup_union_find_kernel";
/// Exact WGSL declaration prefix that must expose [`ENTRY_POINT`].
pub const ENTRY_POINT_DECL: &str =
    "@compute @workgroup_size(64, 1, 1)\nfn workgroup_union_find_kernel";

/// Human-readable backend family used by diagnostics and fixture names.
pub const BACKEND_FAMILY: &str = "workgroup";

/// Workgroup-local compute shader source for this intrinsic.
pub const WGSL: &str = include_str!("union_find.wgsl");

/// Return the shader source passed to the WGSL backend loader.
#[must_use]
pub const fn shader_source() -> &'static str {
    WGSL
}

/// Return the backend family used by diagnostics and fixture names.
#[must_use]
pub const fn backend_family() -> &'static str {
    BACKEND_FAMILY
}

/// Verify that the bundled shader still exposes the registered entry point.
#[must_use]
pub fn exposes_registered_entry_point() -> bool {
    WGSL.contains(ENTRY_POINT_DECL)
}

#[cfg(test)]
mod tests {
    use super::exposes_registered_entry_point;

    #[test]
    fn entry_point_matches_compute_declaration() {
        assert!(exposes_registered_entry_point());
    }
}