remove debug-level print info

Author: hczphn

expander_compiler/src/zkcuda/cpu_monitor.rs

@@ -1,5 +1,4 @@
 pub mod context;
-pub mod cpu_monitor;
 pub mod kernel;
 pub mod mpi_mem_share;
 pub mod proving_system;

expander_compiler/src/zkcuda/mod.rs

@@ -7,7 +7,6 @@ use gkr_engine::{
 };
 use polynomials::RefMultiLinearPoly;
 use serdes::ExpSerde;
-use std::time::Instant;
 use sumcheck::ProverScratchPad;
 
 use crate::{
@@ -18,28 +17,6 @@ use crate::{
     },
 };
 
-/// 获取当前进程的内存使用情况 (RSS, 单位: KB)
-fn get_memory_usage_kb() -> Option<u64> {
-    #[cfg(target_os = "linux")]
-    {
-        if let Ok(content) = std::fs::read_to_string("/proc/self/statm") {
-            let parts: Vec<&str> = content.split_whitespace().collect();
-            if parts.len() >= 2 {
-                // statm 第二个字段是 RSS (以页为单位)
-                // Linux 页大小通常是 4KB
-                if let Ok(rss_pages) = parts[1].parse::<u64>() {
-                    return Some(rss_pages * 4); // 转换为 KB
-                }
-            }
-        }
-        None
-    }
-    #[cfg(not(target_os = "linux"))]
-    {
-        None
-    }
-}
-
 /// ECCCircuit -> ExpanderCircuit
 /// Returns an additional prover scratch pad for later use in GKR.
 pub fn prepare_expander_circuit<F, ECCConfig>(
@@ -51,156 +28,12 @@ where
     ECCConfig: Config,
     ECCConfig::FieldConfig: FieldEngine<CircuitField = F::CircuitField>,
 {
-    // 记录开始时间
-    let start_time = Instant::now();
-    eprintln!("[prepare_expander_circuit] ============== Start ==============");
-
-    // 记录开始时的内存
-    let mem_before = get_memory_usage_kb();
-    eprintln!(
-        "[prepare_expander_circuit] Memory before: {:?} KB",
-        mem_before
-    );
-
-    // Step 1: export_to_expander().flatten()
-    let step1_start = Instant::now();
     let mut expander_circuit = kernel.layered_circuit().export_to_expander().flatten();
-    let step1_duration = step1_start.elapsed();
-    eprintln!(
-        "[prepare_expander_circuit] Step 1 (export_to_expander + flatten) took: {:.3}s",
-        step1_duration.as_secs_f64()
-    );
-
-    // Step 2: 打印电路大小信息
-    let step2_start = Instant::now();
-    let num_layers = expander_circuit.layers.len();
-    let mut total_gates = 0usize;
-    let mut total_add_gates = 0usize;
-    let mut total_mul_gates = 0usize;
-    let mut total_const_gates = 0usize;
-
-    for layer in expander_circuit.layers.iter() {
-        total_add_gates += layer.add.len();
-        total_mul_gates += layer.mul.len();
-        total_const_gates += layer.const_.len();
-        total_gates += layer.add.len() + layer.mul.len() + layer.const_.len();
-    }
-
-    eprintln!("[prepare_expander_circuit] Circuit stats:");
-    eprintln!("  - num_layers: {}", num_layers);
-    eprintln!("  - total_gates: {}", total_gates);
-    eprintln!("  - total_add_gates: {}", total_add_gates);
-    eprintln!("  - total_mul_gates: {}", total_mul_gates);
-    eprintln!("  - total_const_gates: {}", total_const_gates);
-    eprintln!("  - log_input_size: {}", expander_circuit.log_input_size());
-    let step2_duration = step2_start.elapsed();
-    eprintln!(
-        "[prepare_expander_circuit] Step 2 (circuit stats calculation) took: {:.3}s",
-        step2_duration.as_secs_f64()
-    );
-
-    // 记录 export_to_expander().flatten() 后的内存
-    let mem_after_flatten = get_memory_usage_kb();
-    eprintln!(
-        "[prepare_expander_circuit] Memory after flatten: {:?} KB",
-        mem_after_flatten
-    );
-    if let (Some(before), Some(after)) = (mem_before, mem_after_flatten) {
-        eprintln!(
-            "[prepare_expander_circuit] Memory delta (flatten): {} KB ({:.2} MB)",
-            after as i64 - before as i64,
-            (after as i64 - before as i64) as f64 / 1024.0
-        );
-    }
-
-    // Step 3: pre_process_gkr
-    let step3_start = Instant::now();
     expander_circuit.pre_process_gkr();
-    let step3_duration = step3_start.elapsed();
-    eprintln!(
-        "[prepare_expander_circuit] Step 3 (pre_process_gkr) took: {:.3}s",
-        step3_duration.as_secs_f64()
-    );
-
+    
     let (max_num_input_var, max_num_output_var) = super::utils::max_n_vars(&expander_circuit);
-    eprintln!("  - max_num_input_var: {}", max_num_input_var);
-    eprintln!("  - max_num_output_var: {}", max_num_output_var);
-
-    // 记录 pre_process_gkr 后的内存
-    let mem_after_preprocess = get_memory_usage_kb();
-    eprintln!(
-        "[prepare_expander_circuit] Memory after pre_process_gkr: {:?} KB",
-        mem_after_preprocess
-    );
-    if let (Some(before), Some(after)) = (mem_after_flatten, mem_after_preprocess) {
-        eprintln!(
-            "[prepare_expander_circuit] Memory delta (pre_process_gkr): {} KB ({:.2} MB)",
-            after as i64 - before as i64,
-            (after as i64 - before as i64) as f64 / 1024.0
-        );
-    }
-
-    // Step 4: create ProverScratchPad
-    let step4_start = Instant::now();
     let prover_scratch =
         ProverScratchPad::<F>::new(max_num_input_var, max_num_output_var, mpi_world_size);
-    let step4_duration = step4_start.elapsed();
-    eprintln!(
-        "[prepare_expander_circuit] Step 4 (create ProverScratchPad) took: {:.3}s",
-        step4_duration.as_secs_f64()
-    );
-
-    // 记录分配 ProverScratchPad 后的内存
-    let mem_after_scratch = get_memory_usage_kb();
-    eprintln!(
-        "[prepare_expander_circuit] Memory after ProverScratchPad: {:?} KB",
-        mem_after_scratch
-    );
-    if let (Some(before), Some(after)) = (mem_after_preprocess, mem_after_scratch) {
-        eprintln!(
-            "[prepare_expander_circuit] Memory delta (ProverScratchPad): {} KB ({:.2} MB)",
-            after as i64 - before as i64,
-            (after as i64 - before as i64) as f64 / 1024.0
-        );
-    }
-
-    // 总内存增量
-    if let (Some(before), Some(after)) = (mem_before, mem_after_scratch) {
-        eprintln!(
-            "[prepare_expander_circuit] Total memory delta: {} KB ({:.2} MB)",
-            after as i64 - before as i64,
-            (after as i64 - before as i64) as f64 / 1024.0
-        );
-    }
-
-    // 总时间统计
-    let total_duration = start_time.elapsed();
-    eprintln!("[prepare_expander_circuit] ============== Summary ==============");
-    eprintln!(
-        "[prepare_expander_circuit] Step 1 (export + flatten): {:.3}s ({:.1}%)",
-        step1_duration.as_secs_f64(),
-        step1_duration.as_secs_f64() / total_duration.as_secs_f64() * 100.0
-    );
-    eprintln!(
-        "[prepare_expander_circuit] Step 2 (circuit stats):    {:.3}s ({:.1}%)",
-        step2_duration.as_secs_f64(),
-        step2_duration.as_secs_f64() / total_duration.as_secs_f64() * 100.0
-    );
-    eprintln!(
-        "[prepare_expander_circuit] Step 3 (pre_process_gkr):  {:.3}s ({:.1}%)",
-        step3_duration.as_secs_f64(),
-        step3_duration.as_secs_f64() / total_duration.as_secs_f64() * 100.0
-    );
-    eprintln!(
-        "[prepare_expander_circuit] Step 4 (scratch pad):      {:.3}s ({:.1}%)",
-        step4_duration.as_secs_f64(),
-        step4_duration.as_secs_f64() / total_duration.as_secs_f64() * 100.0
-    );
-    eprintln!(
-        "[prepare_expander_circuit] Total time:                {:.3}s",
-        total_duration.as_secs_f64()
-    );
-    eprintln!("[prepare_expander_circuit] ====================================");
 
     (expander_circuit, prover_scratch)
 }