nvtx profiling as feature

ingonyama-zk · Mar 29, 2023 · c49dbb1 · c49dbb1
1 parent bed87f0
commit c49dbb1
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 0 deletions.
diff --git a/fast-danksharding/Cargo.toml b/fast-danksharding/Cargo.toml
@@ -15,6 +15,7 @@ ark-ff = "0.3.0"
 ark-poly = "0.3.0"
 ark-ec = { version = "0.3.0", features = [ "parallel" ] }
 ark-bls12-381 = { version = "0.3.0", optional = true }
+nvtx = { version="1.1.1", optional = true }
 
 [build-dependencies]
 cc = { version = "1.0", features = ["parallel"] }

diff --git a/fast-danksharding/src/fast_danksharding.rs b/fast-danksharding/src/fast_danksharding.rs
@@ -4,6 +4,9 @@ use icicle_utils::{field::Point, *};
 
 use crate::{matrix::*, utils::*, *};
 
+#[cfg(feature = "nvtx")]
+use nvtx::{range_pop, range_push};
+
 pub const FLOW_SIZE: usize = 1 << 12; //4096  //prod flow size
 pub const TEST_SIZE_DIV: usize = 1; //TODO: Prod size / test size for speedup
 pub const TEST_SIZE: usize = FLOW_SIZE / TEST_SIZE_DIV; //test flow size