Skip to content
This repository has been archived by the owner on Aug 16, 2024. It is now read-only.

Commit

Permalink
remove nvtx
Browse files Browse the repository at this point in the history
  • Loading branch information
mcarilli committed Dec 20, 2023
1 parent 762bb26 commit 5504d83
Show file tree
Hide file tree
Showing 3 changed files with 1 addition and 11 deletions.
2 changes: 0 additions & 2 deletions Cargo.toml
Original file line number Diff line number Diff line change
Expand Up @@ -29,8 +29,6 @@ derivative = "*"
bincode = "*"
serde = { version = "1.0", features = ["derive"] }

nvtx = "1.2"


[dev-dependencies]
serial_test = "^2"
Expand Down
4 changes: 0 additions & 4 deletions src/poly.rs
Original file line number Diff line number Diff line change
Expand Up @@ -22,8 +22,6 @@ pub(crate) struct PrecomputedBasisForBarycentric {
pub(crate) bases: DVec<F>,
}

use nvtx::{range_pop, range_push};

impl PrecomputedBasisForBarycentric {
pub fn precompute(domain_size: usize, point: EF) -> CudaResult<Self> {
let mut bases = dvec!(2 * domain_size);
Expand Down Expand Up @@ -290,9 +288,7 @@ impl<'a, P: PolyForm> ComplexPoly<'a, P> {

impl<'a> Poly<'a, LDE> {
pub fn intt(mut self) -> CudaResult<Poly<'a, MonomialBasis>> {
range_push!("Poly<LDE> intt");
ntt::lde_intt(self.storage.as_mut())?;
range_pop!();
Ok(Poly {
storage: self.storage,
marker: std::marker::PhantomData,
Expand Down
6 changes: 1 addition & 5 deletions src/prover.rs
Original file line number Diff line number Diff line change
Expand Up @@ -32,8 +32,6 @@ use crate::{

use super::*;

use nvtx::{range_pop, range_push};

pub fn gpu_prove_from_external_witness_data<
P: boojum::field::traits::field_like::PrimeFieldLikeVectorized<Base = F>,
TR: Transcript<F, CompatibleCap = [F; 4]>,
Expand Down Expand Up @@ -745,10 +743,8 @@ fn gpu_prove_from_trace<
)?;
}

range_push!("quotient evals to monomial");
let quotient_monomial = quotient.intt()?;
range_pop!();
// quotient memory is guaranteed to allow batch ntts for cosets of the quotinet parts
// quotient memory is guaranteed to allow batch ntts for cosets of the quotient parts
let quotient_chunks = quotient_monomial.clone().into_degree_n_polys(domain_size)?;

let quotient_monomial_storage = GenericComplexPolynomialStorage {
Expand Down

0 comments on commit 5504d83

Please sign in to comment.