Split ShardRam EC tree circuit

Author: hero78119

ceno_zkvm/src/e2e.rs

@@ -17,8 +17,8 @@ use crate::{
         ZKVMWitnesses,
     },
     tables::{
-        MemFinalRecord, MemInitRecord, ProgramTableCircuit, ProgramTableConfig, ShardRamCircuit,
-        TableCircuit,
+        MemFinalRecord, MemInitRecord, ProgramTableCircuit, ProgramTableConfig,
+        ShardRamEcTreeCircuit, TableCircuit,
     },
 };
 use ceno_emul::{
@@ -27,8 +27,6 @@ use ceno_emul::{
     StepCellExtractor, StepIndex, StepRecord, SyscallWitness, Tracer, VM_REG_COUNT, VMState,
     WORD_SIZE, Word, WordAddr, host_utils::read_all_messages,
 };
-#[cfg(feature = "gpu")]
-use ceno_gpu::CudaHal;
 use clap::ValueEnum;
 use either::Either;
 use ff_ext::{ExtensionField, SmallField};
@@ -1692,20 +1690,21 @@ pub fn generate_witness<'a, E: ExtensionField>(
                     )
             }).unwrap();
 
-            if let Some(shard_ram_witnesses) =
-                zkvm_witness.get_witness(&ShardRamCircuit::<E>::name())
+            if let Some(shard_ram_ec_tree_witnesses) =
+                zkvm_witness.get_witness(&ShardRamEcTreeCircuit::<E>::name())
             {
                 info_span!("shard_ram_ec_sum").in_scope(|| {
-                    let shard_ram_ec_sum: SepticPoint<E::BaseField> = shard_ram_witnesses
-                        .iter()
-                        .filter(|shard_ram_witness| shard_ram_witness.num_instances[0] > 0)
-                        .map(|shard_ram_witness| {
-                            ShardRamCircuit::<E>::extract_ec_sum(
-                                &system_config.mmu_config.ram_bus_circuit,
-                                &shard_ram_witness.witness_rmms[0],
-                            )
-                        })
-                        .sum();
+                    let shard_ram_ec_sum: SepticPoint<E::BaseField> =
+                        shard_ram_ec_tree_witnesses
+                            .iter()
+                            .filter(|shard_ram_witness| shard_ram_witness.num_instances[0] > 0)
+                            .map(|shard_ram_witness| {
+                                ShardRamEcTreeCircuit::<E>::extract_ec_sum(
+                                    &system_config.mmu_config.ram_bus_ec_tree_circuit,
+                                    &shard_ram_witness.witness_rmms[0],
+                                )
+                            })
+                            .sum();
 
                     let xy = shard_ram_ec_sum
                         .x