Struct GpuDevice 

pub struct GpuDevice {
    pub config: GpuDeviceConfig,
    pub device_ctx: GpuDeviceCtx,
}

Stream-owning device handle. Wraps GpuDeviceConfig with a GpuDeviceCtx that carries the explicit CUDA stream.

Fields

config: GpuDeviceConfig

device_ctx: GpuDeviceCtx

Implementations

impl GpuDevice

pub fn new(params: SystemParams) -> Result<Self, CudaError>

pub fn params(&self) -> &SystemParams

pub fn prover_config(&self) -> &GpuProverConfig

pub fn prover_config_mut(&mut self) -> &mut GpuProverConfig

pub fn sm_count(&self) -> u32

pub fn id(&self) -> u32

pub fn with_cache_rs_code_matrix(self, cache_rs_code_matrix: bool) -> Self

pub fn set_cache_rs_code_matrix(&mut self, cache_rs_code_matrix: bool)

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<HS: GpuHashScheme> DeviceDataTransporter<<HS as GpuHashScheme>::SC, GenericGpuBackend<HS>> for GpuDevice

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

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

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

fn transport_pcs_data_to_device( &self, pcs_data: &StackedPcsData<F, HS::Digest>, ) -> StackedPcsDataGpu<F, HS::Digest>

The commitment and prover_data are assumed to have been previously computed from the trace.

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

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

fn transport_committed_trace_data_to_device( &self, committed_trace: &CommittedTraceData<CpuColMajorBackend<SC>>, ) -> CommittedTraceData<PB>

fn transport_proving_ctx_to_device( &self, ctx: &ProvingContext<CpuColMajorBackend<SC>>, ) -> ProvingContext<PB>

impl<HS: GpuHashScheme, TS: GpuFiatShamirTranscript<HS::SC>> MultiRapProver<GenericGpuBackend<HS>, TS> for GpuDevice

type Artifacts = Vec<BinomialExtensionField<MontyField31<BabyBearParameters>, 4>>

The random opening point r where the batch constraint sumcheck reduces to evaluation claims of trace matrices T, T_{rot} at r_{n_T}.

type PartialProof = (GkrProof<<HS as GpuHashScheme>::SC>, BatchConstraintProof<<HS as GpuHashScheme>::SC>)

The partial proof is the proof that the trace matrices satisfy all constraints assuming that certain polynomial opening claims are validated. In other words, it is a proof that reduces the constraint satisfaction claim to certain polynomial opening claims.

type Error = ProverError

fn prove_rap_constraints( &self, transcript: &mut TS, mpk: &DeviceMultiStarkProvingKey<GenericGpuBackend<HS>>, ctx: &ProvingContext<GenericGpuBackend<HS>>, _common_main_pcs_data: &StackedPcsDataGpu<F, HS::Digest>, ) -> Result<((GkrProof<HS::SC>, BatchConstraintProof<HS::SC>), Vec<EF>), Self::Error>

impl<HS: GpuHashScheme, TS: GpuFiatShamirTranscript<HS::SC>> OpeningProver<GenericGpuBackend<HS>, TS> for GpuDevice

where HS::MerkleHash: MerkleTreeConstructor, HS::Digest: MerkleProofQueryDigest,

type OpeningPoints = Vec<BinomialExtensionField<MontyField31<BabyBearParameters>, 4>>

The shared vector r where each trace matrix T, T_{rot} is opened at r_{n_T}.

type OpeningProof = (StackingProof<<HS as GpuHashScheme>::SC>, WhirProof<<HS as GpuHashScheme>::SC>)

PCS opening proof on host. This should not be a reference.

type Error = ProverError

Computes the opening proof. The common_main_pcs_data is the PcsData for the collection of common main trace matrices. It is owned by the function and may be mutated. The pre_cached_pcs_data_per_commit is the PcsData for the preprocessed and cached trace matrices. These are specified by their PcsData per commitment.

fn prove_openings( &self, transcript: &mut TS, mpk: &DeviceMultiStarkProvingKey<GenericGpuBackend<HS>>, ctx: ProvingContext<GenericGpuBackend<HS>>, common_main_pcs_data: StackedPcsDataGpu<F, HS::Digest>, r: Vec<EF>, ) -> Result<Self::OpeningProof, Self::Error>

impl<HS: GpuHashScheme, TS: GpuFiatShamirTranscript<HS::SC>> ProverDevice<GenericGpuBackend<HS>, TS> for GpuDevice

where HS::MerkleHash: MerkleTreeConstructor, HS::Digest: MerkleProofQueryDigest,

type Error = ProverError

type DeviceCtx = GpuDeviceCtx

Device-specific context (e.g., CUDA stream). Unit () for CPU devices.

fn device_ctx(&self) -> &GpuDeviceCtx

impl<HS: GpuHashScheme> TraceCommitter<GenericGpuBackend<HS>> for GpuDevice

where HS::MerkleHash: MerkleTreeConstructor,

type Error = ProverError

fn commit( &self, traces: &[&DeviceMatrix<F>], ) -> Result<(HS::Digest, StackedPcsDataGpu<F, HS::Digest>), Self::Error>