Struct GpuProgram 

pub struct GpuProgram {
    pub source_size: usize,
    pub output_count: usize,
    pub operation_count: usize,
    pub wgsl: String,
    /* private fields */
}

A compiled program on the GPU: the shader pipeline and its constant tables.

Fields

source_size: usize

output_count: usize

operation_count: usize

Operations the shader evaluates per record.

wgsl: String

Implementations

impl GpuProgram

pub fn new( gpu: &Gpu, circuit: &Circuit, roots: &[Wire], ) -> Result<GpuProgram, GpuError>

Compile the operations that roots need into a shader.

pub fn from_program( gpu: &Gpu, program: &TensorProgram, ) -> Result<GpuProgram, GpuError>

pub fn run(&self, gpu: &Gpu, sources: &[f32]) -> Result<Vec<f32>, GpuError>

Evaluate a batch: sources holds one source table after another, and the result holds each record’s outputs in the order of the roots.