shivini 0.155.9

Shvini is a library implementing a GPU-accelerated zkSync prover
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
59
60
61
62
63

use era_cudart::result::CudaResult;
use era_cudart_sys::CudaError;

#[derive(Debug, Clone, Copy, PartialEq, Eq)]
enum DryRunState {
    Stopped,
    Running,
    Failing(CudaError),
}

use DryRunState::*;

static mut DRY_RUN_STATE: DryRunState = Stopped;

pub(crate) fn dry_run_start() {
    unsafe {
        assert_eq!(DRY_RUN_STATE, Stopped);
        DRY_RUN_STATE = Running;
    }
}

pub(crate) fn dry_run_stop() -> CudaResult<()> {
    unsafe {
        assert_ne!(DRY_RUN_STATE, Stopped);
        let state = DRY_RUN_STATE;
        DRY_RUN_STATE = Stopped;
        if let Failing(e) = state {
            Err(e)
        } else {
            Ok(())
        }
    }
}

pub(crate) fn dry_run_fail(error: CudaError) {
    unsafe {
        assert_ne!(DRY_RUN_STATE, Stopped);
        DRY_RUN_STATE = Failing(error);
    }
}

pub(crate) fn is_dry_run() -> CudaResult<bool> {
    unsafe {
        match DRY_RUN_STATE {
            Stopped => Ok(false),
            Running => Ok(true),
            Failing(e) => Err(e),
        }
    }
}

macro_rules! if_not_dry_run {
    ($($t:tt)*) => {
        if !crate::primitives::dry_run::is_dry_run()? {
            $($t)*
        }
        else {
            Ok(())
        }
    };
}

pub(crate) use if_not_dry_run;