proof-stream 0.1.0

Real-time 3D streaming visualization of ZK proof sessions via Rerun
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

//! Helpers for building `ProofEvent::GpuStatus` from device info.

use crate::events::{GpuSnapshot, ProofEvent};

/// Build a `GpuStatus` event from a slice of device snapshots and progress
/// counters.
pub fn gpu_status_event(
    devices: Vec<GpuSnapshot>,
    matmul_done: usize,
    matmul_total: usize,
    layers_done: usize,
    layers_total: usize,
) -> ProofEvent {
    ProofEvent::GpuStatus {
        devices,
        matmul_done,
        matmul_total,
        layers_done,
        layers_total,
    }
}

/// Create a `GpuSnapshot` with unknown utilization (useful when NVML is not
/// available).
pub fn unknown_gpu_snapshot(device_id: usize, name: impl Into<String>) -> GpuSnapshot {
    GpuSnapshot {
        device_id,
        device_name: name.into(),
        utilization: 0.0,
        free_memory_bytes: None,
    }
}