feat: add coz profiler

Cargo.toml

@@ -15,3 +15,8 @@ rust-version = "1.64.0"
 homepage = "https://github.com/EspressoSystems/jellyfish"
 documentation = "https://jellyfish.docs.espressosys.com"
 repository = "https://github.com/EspressoSystems/jellyfish"
+
+# optional, for coz profiler.
+# see <https://github.com/plasma-umass/coz/tree/master/rust>
+[profile.release]
+debug = 1

flake.nix

@@ -86,7 +86,8 @@
             clangStdenv
             llvm_15
           ] ++ lib.optionals stdenv.isDarwin
-            [ darwin.apple_sdk.frameworks.Security ];
+            [ darwin.apple_sdk.frameworks.Security ]
+            ++ lib.optionals stdenv.isLinux [ coz ];
 
           shellHook = ''
             export RUST_BACKTRACE=full

primitives/Cargo.toml

@@ -28,6 +28,7 @@ chacha20poly1305 = { version = "0.10.1", default-features = false, features = [
         "alloc",
         "rand_core",
 ] }
+coz = { version = "0.1", optional = true }
 crypto_kx = { version = "=0.2.0-pre.0", features = ["serde"] }
 derivative = { version = "2", features = ["use_core"] }
 digest = { version = "0.10.1", default-features = false, features = ["alloc"] }

primitives/src/reed_solomon_code/mod.rs

@@ -11,6 +11,8 @@ use ark_ff::{FftField, Field};
 use ark_poly::{EvaluationDomain, Radix2EvaluationDomain};
 use ark_std::{format, string::ToString, vec, vec::Vec};
 use core::borrow::Borrow;
+#[cfg(all(debug_assertions, target_os = "linux"))]
+use coz;
 
 /// Erasure-encode `data` into `data.len() + parity_size` shares.
 ///
@@ -97,6 +99,9 @@ where
         .clone()
         .map(|share| *share.borrow().0.borrow())
         .collect::<Vec<_>>();
+
+    #[cfg(all(debug_assertions, target_os = "linux"))]
+    coz::begin!("computing l(X)");
     // Calculating l(x) = \prod (x - x_i)
     let mut l = vec![F::zero(); data_size + 1];
     l[0] = F::one();
@@ -107,6 +112,12 @@ where
         }
         l[0] = -x[i - 1] * l[0];
     }
+
+    #[cfg(all(debug_assertions, target_os = "linux"))]
+    {
+        coz::end!("computing l(X)");
+        coz::begin!("computing barycentric weight w_i");
+    }
     // Calculate the barycentric weight w_i
     let w = (0..data_size)
         .map(|i| {
@@ -126,6 +137,11 @@ where
             Ok(ret)
         })
         .collect::<Result<Vec<_>, _>>()?;
+    #[cfg(all(debug_assertions, target_os = "linux"))]
+    {
+        coz::end!("computing barycentric weight w_i");
+        coz::begin!("computing f(X)");
+    }
     // Calculate f(x) = \sum_i l_i(x)
     let mut f = vec![F::zero(); data_size];
     // for i in 0..shares.len() {
@@ -140,6 +156,8 @@ where
             f[j] += weight * li[j];
         }
     }
+    #[cfg(all(debug_assertions, target_os = "linux"))]
+    coz::end!("computing f(X)");
     Ok(f)
 }