sp1-prover-types 6.2.2

Type definitions for the SP1 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

use anyhow::{anyhow, Result};

mod artifacts;
pub use artifacts::*;

mod machine;
pub use machine::*;

pub mod cluster {
    tonic::include_proto!("cluster");
}
pub use cluster::*;

pub mod worker {
    tonic::include_proto!("worker");
}
pub use worker::*;

mod utils;
pub use utils::*;

#[rustfmt::skip]
pub mod network_base_types;

impl WorkerType {
    pub fn from_task_type(task_type: TaskType) -> Self {
        match task_type {
            TaskType::Controller
            | TaskType::PlonkWrap
            | TaskType::Groth16Wrap
            | TaskType::UtilVkeyMapController
            | TaskType::ExecuteOnly
            | TaskType::CoreExecute => WorkerType::Cpu,
            TaskType::ProveShard
            | TaskType::RecursionReduce
            | TaskType::RecursionDeferred
            | TaskType::ShrinkWrap
            | TaskType::SetupVkey
            | TaskType::UtilVkeyMapChunk => WorkerType::Gpu,
            TaskType::MarkerDeferredRecord | TaskType::UnspecifiedTaskType => WorkerType::None,
        }
    }
}

impl WorkerTask {
    pub fn data(&self) -> Result<&TaskData> {
        self.data.as_ref().ok_or(anyhow!("no task data"))
    }
}