vyre-driver-wgpu 0.6.5

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
48
49
50
51
52
53
54
55
56
57
58

use crate::{pipeline, WgpuBackend};

/// Progressive staging: `Program -> WgpuIR -> WGSL -> pipeline`.
///
/// `WgpuIR` is the intermediate artifact returned by
/// [`WgpuBackend::compile`]. Each downstream stage (WGSL emission,
/// pipeline creation) is independently cacheable and testable.
pub struct WgpuIR {
    /// Cached pipeline that already embeds the naga::Module, WGSL
    /// shader source, bind-group layout, and workgroup size.
    pub pipeline: pipeline::WgpuPipeline,
}

impl vyre_driver::Executable for WgpuBackend {
    fn dispatch(
        &self,
        program: &vyre_foundation::ir::Program,
        inputs: &[vyre_driver::MemoryRef<'_>],
        config: &vyre_driver::DispatchConfig,
    ) -> Result<Vec<vyre_driver::Memory>, vyre_driver::BackendError> {
        <Self as vyre_driver::VyreBackend>::dispatch_borrowed(self, program, inputs, config)
    }
}

impl WgpuBackend {
    /// Compile a program once for repeated dispatch.
    pub fn compile(
        &self,
        program: &vyre_foundation::ir::Program,
    ) -> Result<WgpuIR, vyre_driver::BackendError> {
        let config = vyre_driver::DispatchConfig::default();
        self.validate_with_cache(program)?;
        let pipeline = crate::pipeline::WgpuPipeline::compile_with_device_queue(
            program,
            &config,
            self.adapter_info.clone(),
            self.enabled_features,
            self.current_device_queue(),
            self.dispatch_arena_snapshot(),
            self.current_persistent_pool(),
            self.pipeline_cache.clone(),
            self.bind_group_layout_cache.clone(),
        )?;
        Ok(WgpuIR {
            pipeline: (*pipeline).clone(),
        })
    }

    /// Dispatch a previously compiled program artifact.
    pub fn dispatch_compiled(
        &self,
        compiled: &WgpuIR,
        inputs: &[vyre_driver::MemoryRef<'_>],
        config: &vyre_driver::DispatchConfig,
    ) -> Result<Vec<vyre_driver::Memory>, vyre_driver::BackendError> {
        vyre_driver::CompiledPipeline::dispatch_borrowed(&compiled.pipeline, inputs, config)
    }
}