sp1-prover 6.0.1

The SP1 prover implementation
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

use std::{
    sync::{
        atomic::{AtomicU64, Ordering},
        Arc,
    },
    time::Duration,
};

use crate::worker::TaskMetadata;

#[derive(Debug, Clone, Default)]
pub struct ProverMetrics {
    permit_ms: Arc<AtomicU64>,
}

impl ProverMetrics {
    pub fn new() -> Self {
        Self { permit_ms: Arc::new(AtomicU64::new(0)) }
    }

    pub fn gpu_time(&self) -> u64 {
        self.permit_ms.load(Ordering::Relaxed)
    }

    pub fn increment_permit_time(&self, time: Duration) {
        let ms = time.as_millis() as u64;
        self.permit_ms.fetch_add(ms, Ordering::Relaxed);
    }

    pub fn to_metadata(&self) -> TaskMetadata {
        let gpu_time = self.permit_ms.load(Ordering::Relaxed);
        TaskMetadata { gpu_time: Some(gpu_time) }
    }
}