Adding GeneralDepolarizingNoise

Author: ciaranra

crates/pecos-engines/examples/run_noisy_circ.rs

@@ -0,0 +1,41 @@
+use pecos_engines::byte_message::ByteMessage;
+use pecos_engines::engines::quantum::StateVecEngine;
+use pecos_engines::errors::QueueError;
+use pecos_engines::{Engine, GeneralDepolarizingNoise};
+use pecos_engines::{EngineSystem, QuantumSystem};
+
+fn main() -> () {
+    let circ = ByteMessage::quantum_operations_builder()
+        .add_h(&[0])
+        .add_cx(&[0], &[1])
+        .add_measurements(&[0], &[0])
+        .add_measurements(&[1], &[1])
+        .build();
+
+    let quantum = Box::new(StateVecEngine::new(2));
+    let noise = Box::new(GeneralDepolarizingNoise::new(0.1, 0.1, 0.1, 0.1));
+
+    let mut system = QuantumSystem::new(noise, quantum);
+
+    // system.set_seed(42).expect("failed to set seed");
+
+    print!("[");
+    for _ in 0..20 {
+        system.reset().expect("failed to reset");
+        let results = system
+            .process_as_system(circ.clone())
+            .expect("failed to process circ");
+        let meas = results
+            .parse_measurements()
+            .expect("failed to parse measurements");
+
+        print!("\"");
+        for &(_, value) in &meas {
+            print!("{value}");
+        }
+        print!("\", ");
+    }
+    print!("]");
+
+    println!();
+}

crates/pecos-engines/src/byte_message/builder.rs

@@ -184,7 +184,7 @@ impl ByteMessageBuilder {
         let header_size = size_of::<QuantumGateHeader>();
         let qubits_size = gate.qubits.len() * size_of::<u32>();
         let params_size = match gate.gate_type {
-            GateType::RZ => size_of::<f64>(),
+            GateType::RZ | GateType::RZZ => size_of::<f64>(),
             GateType::R1XY => 2 * size_of::<f64>(),
             _ => 0,
         };

crates/pecos-engines/src/byte_message/gate_type.rs

@@ -79,6 +79,9 @@ pub struct QuantumGate {
     pub params: Vec<f64>,
     /// Optional result ID for measurement gates
     pub result_id: Option<usize>,
+    /// Whether the gate should have noise applied to it
+    pub noiseless: bool,
+    // TODO: encode noiseless in the byte representation...
 }
 
 impl QuantumGate {
@@ -95,6 +98,7 @@ impl QuantumGate {
             qubits,
             params,
             result_id,
+            noiseless: false,
         }
     }
 
@@ -162,6 +166,21 @@ impl QuantumGate {
     pub fn prep(qubit: usize) -> Self {
         Self::new(GateType::Prep, vec![qubit], vec![], None)
     }
+
+    #[must_use]
+    pub fn set_noiseless(mut self) -> Self {
+        self.noiseless = true;
+        self
+    }
+
+    #[must_use] pub fn set_noisy(mut self) -> Self {
+        self.noiseless = false;
+        self
+    }
+
+    #[must_use] pub fn is_noiseless(&self) -> bool {
+        self.noiseless
+    }
 }
 
 #[cfg(test)]

crates/pecos-engines/src/byte_message/message.rs

@@ -1,8 +1,8 @@
 use crate::byte_message::builder::ByteMessageBuilder;
 use crate::byte_message::gate_type::{GateType, QuantumGate};
 use crate::byte_message::protocol::{
-    BatchHeader, MeasurementHeader, MeasurementResultHeader, MessageHeader, MessageType,
-    QuantumGateHeader, calc_padding,
+    calc_padding, BatchHeader, MeasurementHeader, MeasurementResultHeader, MessageHeader,
+    MessageType, QuantumGateHeader,
 };
 use crate::errors::QueueError;
 use bytemuck::from_bytes;
@@ -802,7 +802,7 @@ impl ByteMessage {
 #[cfg(test)]
 mod tests {
     use super::*;
-    use crate::engines::{Engine, quantum::StateVecEngine};
+    use crate::engines::{quantum::StateVecEngine, Engine};
 
     #[test]
     fn test_bytemap_builder() {

crates/pecos-engines/src/engines/noise.rs

@@ -1,7 +1,9 @@
 pub mod depolarizing;
+pub mod general_depolarizing;
 pub mod pass_through;
 
 pub use depolarizing::DepolarizingNoise;
+pub use general_depolarizing::GeneralDepolarizingNoise;
 pub use pass_through::PassThroughNoise;
 
 use crate::byte_message::ByteMessage;