Struct GpuDevice 

pub struct GpuDevice {
    pub config: GpuConfig,
    pub id: u32,
    /* private fields */
}

Fields

config: GpuConfig

id: u32

Implementations

impl GpuDevice

pub fn commit_trace<LDE: GpuLde>( &self, trace: DeviceMatrix<BabyBear>, ) -> (Vec<u8>, GpuMerkleTree<LDE>)

pub fn commit_traces_with_lde<LDE: GpuLde>( &self, traces_with_shifts: Vec<(DeviceMatrix<BabyBear>, BabyBear)>, log_blowup: usize, ) -> (Vec<u8>, GpuMerkleTree<LDE>)

Commit a trace to a GPU device.

impl GpuDevice

pub fn new(config: GpuConfig, rap_phase_seq: Option<FriLogUpPhaseGpu>) -> Self

pub fn rap_phase_seq(&self) -> &FriLogUpPhaseGpu

pub fn natural_domain_for_degree( &self, degree: usize, ) -> TwoAdicMultiplicativeCoset<BabyBear>

Trait Implementations

impl Clone for GpuDevice

fn clone(&self) -> GpuDevice

Returns a duplicate of the value.

fn clone_from(&mut self, source: &Self)

Performs copy-assignment from source.

impl Debug for GpuDevice

fn fmt(&self, f: &mut Formatter<'_>) -> Result

Formats the value using the given formatter.

impl DeviceDataTransporter<StarkConfig<TwoAdicFriPcs<MontyField31<BabyBearParameters>, Radix2DitParallel<MontyField31<BabyBearParameters>>, MerkleTreeMmcs<<MontyField31<BabyBearParameters> as Field>::Packing, <MontyField31<BabyBearParameters> as Field>::Packing, PaddingFreeSponge<Poseidon2<<MontyField31<BabyBearParameters> as Field>::Packing, Poseidon2ExternalLayerMonty31<BabyBearParameters, openvm_stark_sdk::::config::baby_bear_poseidon2::Perm::{constant#0}>, Poseidon2InternalLayerMonty31<BabyBearParameters, openvm_stark_sdk::::config::baby_bear_poseidon2::Perm::{constant#0}, BabyBearInternalLayerParameters>, openvm_stark_sdk::::config::baby_bear_poseidon2::Perm::{constant#0}, p3_baby_bear::::poseidon2::Poseidon2BabyBear::{constant#0}>, openvm_stark_sdk::::config::baby_bear_poseidon2::Hash::{constant#0}, openvm_stark_sdk::::config::baby_bear_poseidon2::Hash::{constant#1}, openvm_stark_sdk::::config::baby_bear_poseidon2::Hash::{constant#2}>, TruncatedPermutation<Poseidon2<<MontyField31<BabyBearParameters> as Field>::Packing, Poseidon2ExternalLayerMonty31<BabyBearParameters, openvm_stark_sdk::::config::baby_bear_poseidon2::Perm::{constant#0}>, Poseidon2InternalLayerMonty31<BabyBearParameters, openvm_stark_sdk::::config::baby_bear_poseidon2::Perm::{constant#0}, BabyBearInternalLayerParameters>, openvm_stark_sdk::::config::baby_bear_poseidon2::Perm::{constant#0}, p3_baby_bear::::poseidon2::Poseidon2BabyBear::{constant#0}>, 2, openvm_stark_sdk::::config::baby_bear_poseidon2::Compress::{constant#1}, openvm_stark_sdk::::config::baby_bear_poseidon2::Compress::{constant#2}>, openvm_stark_sdk::::config::baby_bear_poseidon2::ValMmcs::{constant#0}>, ExtensionMmcs<MontyField31<BabyBearParameters>, BinomialExtensionField<MontyField31<BabyBearParameters>, 4>, MerkleTreeMmcs<<MontyField31<BabyBearParameters> as Field>::Packing, <MontyField31<BabyBearParameters> as Field>::Packing, PaddingFreeSponge<Poseidon2<<MontyField31<BabyBearParameters> as Field>::Packing, Poseidon2ExternalLayerMonty31<BabyBearParameters, openvm_stark_sdk::::config::baby_bear_poseidon2::Perm::{constant#0}>, Poseidon2InternalLayerMonty31<BabyBearParameters, openvm_stark_sdk::::config::baby_bear_poseidon2::Perm::{constant#0}, BabyBearInternalLayerParameters>, openvm_stark_sdk::::config::baby_bear_poseidon2::Perm::{constant#0}, p3_baby_bear::::poseidon2::Poseidon2BabyBear::{constant#0}>, openvm_stark_sdk::::config::baby_bear_poseidon2::Hash::{constant#0}, openvm_stark_sdk::::config::baby_bear_poseidon2::Hash::{constant#1}, openvm_stark_sdk::::config::baby_bear_poseidon2::Hash::{constant#2}>, TruncatedPermutation<Poseidon2<<MontyField31<BabyBearParameters> as Field>::Packing, Poseidon2ExternalLayerMonty31<BabyBearParameters, openvm_stark_sdk::::config::baby_bear_poseidon2::Perm::{constant#0}>, Poseidon2InternalLayerMonty31<BabyBearParameters, openvm_stark_sdk::::config::baby_bear_poseidon2::Perm::{constant#0}, BabyBearInternalLayerParameters>, openvm_stark_sdk::::config::baby_bear_poseidon2::Perm::{constant#0}, p3_baby_bear::::poseidon2::Poseidon2BabyBear::{constant#0}>, 2, openvm_stark_sdk::::config::baby_bear_poseidon2::Compress::{constant#1}, openvm_stark_sdk::::config::baby_bear_poseidon2::Compress::{constant#2}>, openvm_stark_sdk::::config::baby_bear_poseidon2::ValMmcs::{constant#0}>>>, FriLogUpPhase<MontyField31<BabyBearParameters>, BinomialExtensionField<MontyField31<BabyBearParameters>, 4>, DuplexChallenger<MontyField31<BabyBearParameters>, Poseidon2<<MontyField31<BabyBearParameters> as Field>::Packing, Poseidon2ExternalLayerMonty31<BabyBearParameters, openvm_stark_sdk::::config::baby_bear_poseidon2::Perm::{constant#0}>, Poseidon2InternalLayerMonty31<BabyBearParameters, openvm_stark_sdk::::config::baby_bear_poseidon2::Perm::{constant#0}, BabyBearInternalLayerParameters>, openvm_stark_sdk::::config::baby_bear_poseidon2::Perm::{constant#0}, p3_baby_bear::::poseidon2::Poseidon2BabyBear::{constant#0}>, openvm_stark_sdk::::config::baby_bear_poseidon2::Challenger::{constant#0}, openvm_stark_sdk::::config::baby_bear_poseidon2::Challenger::{constant#1}>>, BinomialExtensionField<MontyField31<BabyBearParameters>, 4>, DuplexChallenger<MontyField31<BabyBearParameters>, Poseidon2<<MontyField31<BabyBearParameters> as Field>::Packing, Poseidon2ExternalLayerMonty31<BabyBearParameters, openvm_stark_sdk::::config::baby_bear_poseidon2::Perm::{constant#0}>, Poseidon2InternalLayerMonty31<BabyBearParameters, openvm_stark_sdk::::config::baby_bear_poseidon2::Perm::{constant#0}, BabyBearInternalLayerParameters>, openvm_stark_sdk::::config::baby_bear_poseidon2::Perm::{constant#0}, p3_baby_bear::::poseidon2::Poseidon2BabyBear::{constant#0}>, openvm_stark_sdk::::config::baby_bear_poseidon2::Challenger::{constant#0}, openvm_stark_sdk::::config::baby_bear_poseidon2::Challenger::{constant#1}>>, GpuBackend> for GpuDevice

fn transport_committed_trace_to_device( &self, commitment: Com<SC>, trace: &Arc<RowMajorMatrix<Val<SC>>>, _: &Arc<PcsProverData<SC>>, ) -> CommittedTraceData<GpuBackend>

We ignore the host prover data because it’s faster to just re-commit on GPU instead of doing H2D transfer.

fn transport_pk_to_device( &self, mpk: &MultiStarkProvingKey<SC>, ) -> DeviceMultiStarkProvingKey<GpuBackend>

Transport the proving key to the device, filtering for only the provided air_ids.

fn transport_matrix_to_device( &self, matrix: &Arc<RowMajorMatrix<F>>, ) -> DeviceMatrix<F>

fn transport_matrix_from_device_to_host( &self, matrix: &DeviceMatrix<F>, ) -> Arc<RowMajorMatrix<F>>

Transport a device matrix to host. This should only be used for testing / debugging purposes.

impl OpeningProver<GpuBackend> for GpuDevice

fn open( &self, challenger: &mut <GpuBackend as ProverBackend>::Challenger, preprocessed: Vec<&<GpuBackend as ProverBackend>::PcsData>, main: Vec<<GpuBackend as ProverBackend>::PcsData>, after_phase: Vec<<GpuBackend as ProverBackend>::PcsData>, quotient_data: <GpuBackend as ProverBackend>::PcsData, quotient_degrees: &[u8], ) -> OpeningProof<PcsProof<SC>, EF>

Opening proof for multiple RAP matrices, where

impl QuotientCommitter<GpuBackend> for GpuDevice

fn eval_and_commit_quotient( &self, challenger: &mut <GpuBackend as ProverBackend>::Challenger, pk_views: &[&DeviceStarkProvingKey<GpuBackend>], public_values: &[Vec<<GpuBackend as ProverBackend>::Val>], cached_pcs_datas_per_air: &[Vec<<GpuBackend as ProverBackend>::PcsData>], common_main_pcs_data: &<GpuBackend as ProverBackend>::PcsData, prover_data_after: &ProverDataAfterRapPhases<GpuBackend>, ) -> (<GpuBackend as ProverBackend>::Commitment, <GpuBackend as ProverBackend>::PcsData)

Given a view of the PCS data from all phases of proving, first get the trace polynomials evaluated on the quotient domains. Then compute the quotient polynomial evaluated on the quotient domain and commit to it.

impl RapPartialProver<GpuBackend> for GpuDevice

fn partially_prove( &self, challenger: &mut <GpuBackend as ProverBackend>::Challenger, mpk: &DeviceMultiStarkProvingKeyView<'_, GpuBackend>, trace_views: Vec<AirView<<GpuBackend as ProverBackend>::Matrix, <GpuBackend as ProverBackend>::Val>>, ) -> (Option<RapPhaseSeqPartialProof<SC>>, ProverDataAfterRapPhases<GpuBackend>)

The trace_views are the respective (owned) trace matrices, evaluated on the trace domain. Currently this function does not provide a view of any already committed data associated with the trace views, although that data is available.

impl TraceCommitter<GpuBackend> for GpuDevice

fn commit(&self, traces: &[DeviceMatrix<F>]) -> (Com<SC>, GpuPcsData)

impl ProverDevice<GpuBackend> for GpuDevice