shivini 0.155.11

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

use super::*;

use boojum_cuda::ops_complex::select;
use era_cudart::slice::DeviceSlice;

pub fn variable_assignment(
    d_variable_indexes: &DVec<u32>,
    d_variable_values: &DVec<F>,
    d_result: &mut [F],
) -> CudaResult<()> {
    if d_variable_indexes.is_empty() {
        return Ok(());
    }
    assert!(d_variable_values.len() > 0);
    assert_eq!(d_variable_indexes.len(), d_result.len());
    let (d_variable_indexes_ref, d_variable_values_ref, d_result) = unsafe {
        (
            DeviceSlice::from_slice(d_variable_indexes),
            DeviceSlice::from_slice(d_variable_values),
            DeviceSlice::from_mut_slice(d_result),
        )
    };
    if_not_dry_run! {
        select(
            d_variable_indexes_ref,
            d_variable_values_ref,
            d_result,
            get_stream(),
        )
    }
}