pyqrack-complex128 1.62.0__py3-none-macosx_13_0_x86_64.whl → 1.71.1__py3-none-macosx_13_0_x86_64.whl

pyqrack/qrack_ace_backend.py

@@ -27,6 +27,169 @@ except ImportError:
     _IS_QISKIT_AER_AVAILABLE = False
 
 
+# Initial stub and concept produced through conversation with Elara
+# (the custom OpenAI GPT)
+class LHVQubit:
+    def __init__(self, toClone=None):
+        # Initial state in "Bloch vector" terms, defaults to |0⟩
+        if toClone:
+            self.bloch = toClone.bloch.copy()
+        else:
+            self.reset()
+
+    def reset(self):
+        self.bloch = [0.0, 0.0, 1.0]
+
+    def h(self):
+        # Hadamard: rotate around Y-axis then X-axis (simplified for LHV)
+        x, y, z = self.bloch
+        self.bloch = [(x + z) / math.sqrt(2), y, (z - x) / math.sqrt(2)]
+
+    def x(self):
+        x, y, z = self.bloch
+        self.bloch = [x, y, -z]
+
+    def y(self):
+        x, y, z = self.bloch
+        self.bloch = [-x, y, z]
+
+    def z(self):
+        x, y, z = self.bloch
+        self.bloch = [x, -y, z]
+
+    def rx(self, theta):
+        # Rotate Bloch vector around X-axis by angle theta
+        x, y, z = self.bloch
+        cos_theta = math.cos(theta)
+        sin_theta = math.sin(theta)
+        new_y = cos_theta * y - sin_theta * z
+        new_z = sin_theta * y + cos_theta * z
+        self.bloch = [x, new_y, new_z]
+
+    def ry(self, theta):
+        # Rotate Bloch vector around Y-axis by angle theta
+        x, y, z = self.bloch
+        cos_theta = math.cos(theta)
+        sin_theta = math.sin(theta)
+        new_x = cos_theta * x + sin_theta * z
+        new_z = -sin_theta * x + cos_theta * z
+        self.bloch = [new_x, y, new_z]
+
+    def rz(self, theta):
+        # Rotate Bloch vector around Z-axis by angle theta (in radians)
+        x, y, z = self.bloch
+        cos_theta = math.cos(theta)
+        sin_theta = math.sin(theta)
+        new_x = cos_theta * x - sin_theta * y
+        new_y = sin_theta * x + cos_theta * y
+        self.bloch = [new_x, new_y, z]
+
+    def s(self):
+        self.rz(math.pi / 2)
+
+    def adjs(self):
+        self.rz(-math.pi / 2)
+
+    def t(self):
+        self.rz(math.pi / 4)
+
+    def adjt(self):
+        self.rz(-math.pi / 4)
+
+    def u(self, theta, phi, lam):
+        # Apply general single-qubit unitary gate
+        self.rz(lam)
+        self.ry(theta)
+        self.rz(phi)
+
+    # Provided verbatim by Elara (the custom OpenAI GPT):
+    def mtrx(self, matrix):
+        """
+        Apply a 2x2 unitary matrix to the LHV Bloch vector using only standard math/cmath.
+        Matrix format: [a, b, c, d] for [[a, b], [c, d]]
+        """
+        a, b, c, d = matrix
+
+        # Current Bloch vector
+        x, y, z = self.bloch
+
+        # Convert to density matrix ρ = ½ (I + xσx + yσy + zσz)
+        rho = [[(1 + z) / 2, (x - 1j * y) / 2], [(x + 1j * y) / 2, (1 - z) / 2]]
+
+        # Compute U * ρ
+        u_rho = [
+            [a * rho[0][0] + b * rho[1][0], a * rho[0][1] + b * rho[1][1]],
+            [c * rho[0][0] + d * rho[1][0], c * rho[0][1] + d * rho[1][1]],
+        ]
+
+        # Compute (U * ρ) * U†
+        rho_prime = [
+            [
+                u_rho[0][0] * a.conjugate() + u_rho[0][1] * b.conjugate(),
+                u_rho[0][0] * c.conjugate() + u_rho[0][1] * d.conjugate(),
+            ],
+            [
+                u_rho[1][0] * a.conjugate() + u_rho[1][1] * b.conjugate(),
+                u_rho[1][0] * c.conjugate() + u_rho[1][1] * d.conjugate(),
+            ],
+        ]
+
+        # Extract Bloch components: Tr(ρ'σi) = 2 * Re[...]
+        new_x = 2 * rho_prime[0][1].real + 2 * rho_prime[1][0].real
+        new_y = 2 * (rho_prime[0][1].imag - rho_prime[1][0].imag)
+        new_z = 2 * rho_prime[0][0].real - 1  # since Tr(ρ') = 1
+
+        p = math.sqrt(new_x**2 + new_y**2 + new_z**2)
+
+        new_x /= p
+        new_y /= p
+        new_z /= p
+
+        self.bloch = [new_x, new_y, new_z]
+
+    def prob(self, basis=Pauli.PauliZ):
+        """Sample a classical outcome from the current 'quantum' state"""
+        if basis == Pauli.PauliZ:
+            prob_1 = (1 - self.bloch[2]) / 2
+        elif basis == Pauli.PauliX:
+            prob_1 = (1 - self.bloch[0]) / 2
+        elif basis == Pauli.PauliY:
+            prob_1 = (1 - self.bloch[1]) / 2
+        else:
+            raise ValueError(f"Unsupported basis: {basis}")
+        return prob_1
+
+    def m(self):
+        result = random.random() < self.prob()
+        self.reset()
+        if result:
+            self.x()
+        return result
+
+
+# Provided by Elara (the custom OpenAI GPT)
+def _cpauli_lhv(prob, targ, axis, anti, theta=math.pi):
+    """
+    Apply a 'soft' controlled-Pauli gate: rotate target qubit
+    proportionally to control's Z expectation value.
+
+    theta: full rotation angle if control in |1⟩
+    """
+    # Control influence is (1 - ctrl.bloch[2]) / 2 = P(|1⟩)
+    # BUT we avoid collapse by using the expectation value:
+    control_influence = (1 - prob) if anti else prob
+
+    effective_theta = control_influence * theta
+
+    # Apply partial rotation to target qubit:
+    if axis == Pauli.PauliX:
+        targ.rx(effective_theta)
+    elif axis == Pauli.PauliY:
+        targ.ry(effective_theta)
+    elif axis == Pauli.PauliZ:
+        targ.rz(effective_theta)
+
+
 class QrackAceBackend:
     """A back end for elided quantum error correction
 
@@ -45,7 +208,6 @@ class QrackAceBackend:
         sim(QrackSimulator): Array of simulators corresponding to "patches" between boundary rows.
         long_range_columns(int): How many ideal rows between QEC boundary rows?
         is_transpose(bool): Rows are long if False, columns are long if True
-        correction_bias(float): Bias magnitude and direction during pseudo-QEC
     """
 
     def __init__(
@@ -54,7 +216,6 @@ class QrackAceBackend:
         long_range_columns=4,
         long_range_rows=4,
         is_transpose=False,
-        correction_bias=0,
         isTensorNetwork=False,
         isSchmidtDecomposeMulti=False,
         isSchmidtDecompose=True,
@@ -83,7 +244,6 @@ class QrackAceBackend:
         self.long_range_columns = long_range_columns
         self.long_range_rows = long_range_rows
         self.is_transpose = is_transpose
-        self.correction_bias = correction_bias
 
         fppow = 5
         if "QRACK_FPPOW" in os.environ:
@@ -122,6 +282,7 @@ class QrackAceBackend:
         self._qubits = []
         sim_counts = [0] * sim_count
         sim_id = 0
+        tot_qubits = 0
         for r in self._is_row_long_range:
             for c in self._is_col_long_range:
                 qubit = [(sim_id, sim_counts[sim_id])]
@@ -132,6 +293,14 @@ class QrackAceBackend:
                     qubit.append((t_sim_id, sim_counts[t_sim_id]))
                     sim_counts[t_sim_id] += 1
 
+                    qubit.append(
+                        LHVQubit(
+                            toClone=(
+                                toClone._qubits[tot_qubits][2] if toClone else None
+                            )
+                        )
+                    )
+
                 if (not c) and (not r):
                     t_sim_id = (sim_id + col_patch_count) % sim_count
                     qubit.append((t_sim_id, sim_counts[t_sim_id]))
@@ -145,6 +314,7 @@ class QrackAceBackend:
                     sim_id = (sim_id + 1) % sim_count
 
                 self._qubits.append(qubit)
+                tot_qubits += 1
 
         self.sim = []
         for i in range(sim_count):
@@ -262,16 +432,34 @@ class QrackAceBackend:
     def _unpack(self, lq):
         return self._qubits[lq]
 
-    def _get_qb_indices(self, hq):
+    def _get_qb_lhv_indices(self, hq):
         qb = []
         if len(hq) < 2:
             qb = [0]
+            lhv = -1
         elif len(hq) < 4:
             qb = [0, 1]
+            lhv = 2
         else:
-            qb = [0, 1, 2, 3]
+            qb = [0, 1, 3, 4]
+            lhv = 2
 
-        return qb
+        return qb, lhv
+
+    def _get_lhv_bloch_angles(self, sim):
+        # Z axis
+        z = sim.bloch[2]
+
+        # X axis
+        x = sim.bloch[0]
+
+        # Y axis
+        y = sim.bloch[1]
+
+        inclination = math.atan2(math.sqrt(x**2 + y**2), z)
+        azimuth = math.atan2(y, x)
+
+        return azimuth, inclination
 
     def _get_bloch_angles(self, hq):
         sim = self.sim[hq[0]].clone()
@@ -315,100 +503,146 @@ class QrackAceBackend:
 
         sim.mtrx([m00, m01, m10, m11], q)
 
+    def _rotate_lhv_to_bloch(self, sim, delta_azimuth, delta_inclination):
+        # Apply rotation as "Azimuth, Inclination" (AI)
+        cosA = math.cos(delta_azimuth)
+        sinA = math.sin(delta_azimuth)
+        cosI = math.cos(delta_inclination / 2)
+        sinI = math.sin(delta_inclination / 2)
+
+        m00 = complex(cosI, 0)
+        m01 = complex(-cosA, sinA) * sinI
+        m10 = complex(cosA, sinA) * sinI
+        m11 = complex(cosI, 0)
+
+        sim.mtrx([m00, m01, m10, m11])
+
     def _correct(self, lq, phase=False, skip_rotation=False):
         hq = self._unpack(lq)
 
         if len(hq) == 1:
             return
 
-        qb = self._get_qb_indices(hq)
+        qb, lhv = self._get_qb_lhv_indices(hq)
 
         if phase:
             for q in qb:
                 b = hq[q]
                 self.sim[b[0]].h(b[1])
+            b = hq[lhv]
+            b.h()
 
         if len(hq) == 5:
             # RMS
             p = [
                 self.sim[hq[0][0]].prob(hq[0][1]),
                 self.sim[hq[1][0]].prob(hq[1][1]),
-                self.sim[hq[2][0]].prob(hq[2][1]),
+                hq[2].prob(),
                 self.sim[hq[3][0]].prob(hq[3][1]),
+                self.sim[hq[4][0]].prob(hq[4][1]),
             ]
             # Balancing suggestion from Elara (the custom OpenAI GPT)
-            prms = math.sqrt((p[0] ** 2 + p[1] ** 2 + p[2] ** 2 + p[3] ** 2) / 4)
+            prms = math.sqrt(
+                (p[0] ** 2 + p[1] ** 2 + 3 * (p[2] ** 2) + p[3] ** 2 + p[4] ** 2) / 7
+            )
             qrms = math.sqrt(
-                ((1 - p[0]) ** 2 + (1 - p[1]) ** 2 + (1 - p[2]) ** 2 + (1 - p[3]) ** 2)
-                / 4
+                (
+                    (1 - p[0]) ** 2
+                    + (1 - p[1]) ** 2
+                    + 3 * ((1 - p[2]) ** 2)
+                    + (1 - p[3]) ** 2
+                    + (1 - p[4]) ** 2
+                )
+                / 7
             )
             result = ((prms + (1 - qrms)) / 2) >= 0.5
             syndrome = (
-                [1 - p[0], 1 - p[1], 1 - p[2], 1 - p[3]]
+                [1 - p[0], 1 - p[1], 1 - p[2], 1 - p[3], 1 - p[4]]
                 if result
-                else [p[0], p[1], p[2], p[3]]
+                else [p[0], p[1], p[2], p[3], p[4]]
             )
-            for q in range(4):
+            for q in range(5):
                 if syndrome[q] > (0.5 + self._epsilon):
-                    self.sim[hq[q][0]].x(hq[q][1])
+                    if q == 2:
+                        hq[q].x()
+                    else:
+                        self.sim[hq[q][0]].x(hq[q][1])
 
             if not skip_rotation:
-                a, i = [0, 0, 0, 0], [0, 0, 0, 0]
+                a, i = [0, 0, 0, 0, 0], [0, 0, 0, 0, 0]
                 a[0], i[0] = self._get_bloch_angles(hq[0])
                 a[1], i[1] = self._get_bloch_angles(hq[1])
-                a[2], i[2] = self._get_bloch_angles(hq[3])
-                a[3], i[3] = self._get_bloch_angles(hq[4])
+                a[2], i[2] = self._get_lhv_bloch_angles(hq[2])
+                a[3], i[3] = self._get_bloch_angles(hq[3])
+                a[4], i[4] = self._get_bloch_angles(hq[4])
 
                 a_target = 0
                 i_target = 0
-                for x in range(4):
+                for x in range(5):
+                    if x == 2:
+                        continue
                     a_target += a[x]
                     i_target += i[x]
 
-                a_target /= 4
-                i_target /= 4
-                for x in range(4):
-                    self._rotate_to_bloch(hq[x], a_target - a[x], i_target - i[x])
-
-                self.apply_magnetic_bias([lq], self.correction_bias)
+                a_target /= 5
+                i_target /= 5
+                for x in range(5):
+                    if x == 2:
+                        self._rotate_lhv_to_bloch(
+                            hq[x], a_target - a[x], i_target - i[x]
+                        )
+                    else:
+                        self._rotate_to_bloch(hq[x], a_target - a[x], i_target - i[x])
 
         else:
             # RMS
             p = [
                 self.sim[hq[0][0]].prob(hq[0][1]),
                 self.sim[hq[1][0]].prob(hq[1][1]),
+                hq[2].prob(),
             ]
             # Balancing suggestion from Elara (the custom OpenAI GPT)
-            prms = math.sqrt((p[0] ** 2 + p[1] ** 2) / 3)
-            qrms = math.sqrt(((1 - p[0]) ** 2 + (1 - p[1]) ** 2) / 3)
+            prms = math.sqrt((p[0] ** 2 + p[1] ** 2 + p[2] ** 2) / 3)
+            qrms = math.sqrt(((1 - p[0]) ** 2 + (1 - p[1]) ** 2 + (1 - p[2]) ** 2) / 3)
             result = ((prms + (1 - qrms)) / 2) >= 0.5
-            syndrome = [1 - p[0], 1 - p[1]] if result else [p[0], p[1]]
-            for q in range(2):
+            syndrome = [1 - p[0], 1 - p[1], 1 - p[2]] if result else [p[0], p[1], p[2]]
+            for q in range(3):
                 if syndrome[q] > (0.5 + self._epsilon):
-                    self.sim[hq[q][0]].x(hq[q][1])
+                    if q == 2:
+                        hq[q].x()
+                    else:
+                        self.sim[hq[q][0]].x(hq[q][1])
 
             if not skip_rotation:
-                a, i = [0, 0], [0, 0]
+                a, i = [0, 0, 0], [0, 0, 0]
                 a[0], i[0] = self._get_bloch_angles(hq[0])
                 a[1], i[1] = self._get_bloch_angles(hq[1])
+                a[2], i[2] = self._get_lhv_bloch_angles(hq[2])
 
                 a_target = 0
                 i_target = 0
-                for x in range(2):
+                for x in range(3):
+                    if x == 2:
+                        continue
                     a_target += a[x]
                     i_target += i[x]
 
-                a_target /= 2
-                i_target /= 2
-                for x in range(2):
-                    self._rotate_to_bloch(hq[x], a_target - a[x], i_target - i[x])
-
-                self.apply_magnetic_bias([lq], self.correction_bias)
+                a_target /= 3
+                i_target /= 3
+                for x in range(3):
+                    if x == 2:
+                        self._rotate_lhv_to_bloch(
+                            hq[x], a_target - a[x], i_target - i[x]
+                        )
+                    else:
+                        self._rotate_to_bloch(hq[x], a_target - a[x], i_target - i[x])
 
         if phase:
             for q in qb:
                 b = hq[q]
                 self.sim[b[0]].h(b[1])
+            b = hq[lhv]
+            b.h()
 
     def apply_magnetic_bias(self, q, b):
         if b == 0:
@@ -416,11 +650,22 @@ class QrackAceBackend:
         b = math.exp(b)
         for x in q:
             hq = self._unpack(x)
-            for h in hq:
-                a, i = self._get_bloch_angles(h)
-                self._rotate_to_bloch(
-                    h, math.atan(math.tan(a) * b) - a, math.atan(math.tan(i) * b) - i
-                )
+            for c in range(len(hq)):
+                h = hq[c]
+                if c == 2:
+                    a, i = self._get_lhv_bloch_angles(h)
+                    self._rotate_lhv_to_bloch(
+                        h,
+                        math.atan(math.tan(a) * b) - a,
+                        math.atan(math.tan(i) * b) - i,
+                    )
+                else:
+                    a, i = self._get_bloch_angles(h)
+                    self._rotate_to_bloch(
+                        h,
+                        math.atan(math.tan(a) * b) - a,
+                        math.atan(math.tan(i) * b) - i,
+                    )
 
     def u(self, lq, th, ph, lm):
         hq = self._unpack(lq)
@@ -429,12 +674,15 @@ class QrackAceBackend:
             self.sim[b[0]].u(b[1], th, ph, lm)
             return
 
-        qb = self._get_qb_indices(hq)
+        qb, lhv = self._get_qb_lhv_indices(hq)
 
         for q in qb:
             b = hq[q]
             self.sim[b[0]].u(b[1], th, ph, lm)
 
+        b = hq[lhv]
+        b.u(th, ph, lm)
+
         self._correct(lq, False, True)
         self._correct(lq, True, False)
 
@@ -445,12 +693,20 @@ class QrackAceBackend:
             self.sim[b[0]].r(p, th, b[1])
             return
 
-        qb = self._get_qb_indices(hq)
+        qb, lhv = self._get_qb_lhv_indices(hq)
 
         for q in qb:
             b = hq[q]
             self.sim[b[0]].r(p, th, b[1])
 
+        b = hq[lhv]
+        if p == Pauli.PauliX:
+            b.rx(th)
+        elif p == Pauli.PauliY:
+            b.ry(th)
+        elif p == Pauli.PauliZ:
+            b.rz(th)
+
         if p != Pauli.PauliZ:
             self._correct(lq, False, p != Pauli.PauliX)
         if p != Pauli.PauliX:
@@ -465,12 +721,15 @@ class QrackAceBackend:
 
         self._correct(lq)
 
-        qb = self._get_qb_indices(hq)
+        qb, lhv = self._get_qb_lhv_indices(hq)
 
         for q in qb:
             b = hq[q]
             self.sim[b[0]].h(b[1])
 
+        b = hq[lhv]
+        b.h()
+
         self._correct(lq)
 
     def s(self, lq):
@@ -480,12 +739,15 @@ class QrackAceBackend:
             self.sim[b[0]].s(b[1])
             return
 
-        qb = self._get_qb_indices(hq)
+        qb, lhv = self._get_qb_lhv_indices(hq)
 
         for q in qb:
             b = hq[q]
             self.sim[b[0]].s(b[1])
 
+        b = hq[lhv]
+        b.s()
+
     def adjs(self, lq):
         hq = self._unpack(lq)
         if len(hq) < 2:
@@ -493,12 +755,15 @@ class QrackAceBackend:
             self.sim[b[0]].adjs(b[1])
             return
 
-        qb = self._get_qb_indices(hq)
+        qb, lhv = self._get_qb_lhv_indices(hq)
 
         for q in qb:
             b = hq[q]
             self.sim[b[0]].adjs(b[1])
 
+        b = hq[lhv]
+        b.adjs()
+
     def x(self, lq):
         hq = self._unpack(lq)
         if len(hq) < 2:
@@ -506,12 +771,15 @@ class QrackAceBackend:
             self.sim[b[0]].x(b[1])
             return
 
-        qb = self._get_qb_indices(hq)
+        qb, lhv = self._get_qb_lhv_indices(hq)
 
         for q in qb:
             b = hq[q]
             self.sim[b[0]].x(b[1])
 
+        b = hq[lhv]
+        b.x()
+
     def y(self, lq):
         hq = self._unpack(lq)
         if len(hq) < 2:
@@ -519,12 +787,15 @@ class QrackAceBackend:
             self.sim[b[0]].y(b[1])
             return
 
-        qb = self._get_qb_indices(hq)
+        qb, lhv = self._get_qb_lhv_indices(hq)
 
         for q in qb:
             b = hq[q]
             self.sim[b[0]].y(b[1])
 
+        b = hq[lhv]
+        b.y()
+
     def z(self, lq):
         hq = self._unpack(lq)
         if len(hq) < 2:
@@ -532,12 +803,15 @@ class QrackAceBackend:
             self.sim[b[0]].z(b[1])
             return
 
-        qb = self._get_qb_indices(hq)
+        qb, lhv = self._get_qb_lhv_indices(hq)
 
         for q in qb:
             b = hq[q]
             self.sim[b[0]].z(b[1])
 
+        b = hq[lhv]
+        b.z()
+
     def t(self, lq):
         hq = self._unpack(lq)
         if len(hq) < 2:
@@ -545,12 +819,15 @@ class QrackAceBackend:
             self.sim[b[0]].t(b[1])
             return
 
-        qb = self._get_qb_indices(hq)
+        qb, lhv = self._get_qb_lhv_indices(hq)
 
         for q in qb:
             b = hq[q]
             self.sim[b[0]].t(b[1])
 
+        b = hq[lhv]
+        b.t()
+
     def adjt(self, lq):
         hq = self._unpack(lq)
         if len(hq) < 2:
@@ -558,12 +835,15 @@ class QrackAceBackend:
             self.sim[b[0]].adjt(b[1])
             return
 
-        qb = self._get_qb_indices(hq)
+        qb, lhv = self._get_qb_lhv_indices(hq)
 
         for q in qb:
             b = hq[q]
             self.sim[b[0]].adjt(b[1])
 
+        b = hq[lhv]
+        b.adjt()
+
     def _get_gate(self, pauli, anti, sim_id):
         gate = None
         shadow = None
@@ -604,11 +884,15 @@ class QrackAceBackend:
 
         return connected, boundary
 
-    def _apply_coupling(self, pauli, anti, qb1, hq1, qb2, hq2, lq1_lr):
+    def _apply_coupling(self, pauli, anti, qb1, lhv1, hq1, qb2, lhv2, hq2, lq1_lr):
         for q1 in qb1:
+            if q1 == lhv1:
+                continue
             b1 = hq1[q1]
             gate_fn, shadow_fn = self._get_gate(pauli, anti, b1[0])
             for q2 in qb2:
+                if q2 == lhv2:
+                    continue
                 b2 = hq2[q2]
                 if b1[0] == b2[0]:
                     gate_fn([b1[1]], b2[1])
@@ -633,10 +917,18 @@ class QrackAceBackend:
 
         self._correct(lq1)
 
-        qb1 = self._get_qb_indices(hq1)
-        qb2 = self._get_qb_indices(hq2)
+        qb1, lhv1 = self._get_qb_lhv_indices(hq1)
+        qb2, lhv2 = self._get_qb_lhv_indices(hq2)
         # Apply cross coupling on hardware qubits first
-        self._apply_coupling(pauli, anti, qb1, hq1, qb2, hq2, lq1_lr)
+        self._apply_coupling(pauli, anti, qb1, lhv1, hq1, qb2, lhv2, hq2, lq1_lr)
+        # Apply coupling to the local-hidden-variable target
+        if lhv2 >= 0:
+            _cpauli_lhv(
+                hq1[lhv1].prob() if lhv1 >= 0 else self.sim[hq1[0][0]].prob(hq1[0][1]),
+                hq2[lhv2],
+                pauli,
+                anti,
+            )
 
         self._correct(lq1, True)
         if pauli != Pauli.PauliZ:
@@ -738,20 +1030,29 @@ class QrackAceBackend:
                 self.sim[hq[4][0]].prob(hq[4][1]),
             ]
             # Balancing suggestion from Elara (the custom OpenAI GPT)
-            prms = math.sqrt((p[0] ** 2 + p[1] ** 2 + p[2] ** 2 + p[3] ** 2) / 4)
+            prms = math.sqrt(
+                (p[0] ** 2 + p[1] ** 2 + 3 * (p[2] ** 2) + p[3] ** 2 + p[4] ** 2) / 7
+            )
             qrms = math.sqrt(
-                ((1 - p[0]) ** 2 + (1 - p[1]) ** 2 + (1 - p[2]) ** 2 + (1 - p[3]) ** 2)
-                / 4
+                (
+                    (1 - p[0]) ** 2
+                    + (1 - p[1]) ** 2
+                    + 3 * ((1 - p[2]) ** 2)
+                    + (1 - p[3]) ** 2
+                    + (1 - p[4]) ** 2
+                )
+                / 7
             )
         else:
             # RMS
             p = [
                 self.sim[hq[0][0]].prob(hq[0][1]),
                 self.sim[hq[1][0]].prob(hq[1][1]),
+                hq[2].prob(),
             ]
             # Balancing suggestion from Elara (the custom OpenAI GPT)
-            prms = math.sqrt((p[0] ** 2 + p[1] ** 2) / 3)
-            qrms = math.sqrt(((1 - p[0]) ** 2 + (1 - p[1]) ** 2) / 3)
+            prms = math.sqrt((p[0] ** 2 + p[1] ** 2 + p[2] ** 2) / 3)
+            qrms = math.sqrt(((1 - p[0]) ** 2 + (1 - p[1]) ** 2 + (1 - p[2]) ** 2) / 3)
 
         return (prms + (1 - qrms)) / 2
 
@@ -764,7 +1065,7 @@ class QrackAceBackend:
         p = self.prob(lq)
         result = ((p + self._epsilon) >= 1) or (random.random() < p)
 
-        qb = self._get_qb_indices(hq)
+        qb, lhv = self._get_qb_lhv_indices(hq)
 
         for q in qb:
             b = hq[q]
@@ -775,6 +1076,11 @@ class QrackAceBackend:
             else:
                 self.sim[b[0]].force_m(b[1], result)
 
+        b = hq[lhv]
+        b.reset()
+        if result:
+            b.x()
+
         return result
 
     def force_m(self, lq, result):
@@ -785,7 +1091,7 @@ class QrackAceBackend:
 
         self._correct(lq)
 
-        qb = self._get_qb_indices(hq)
+        qb, lhv = self._get_qb_lhv_indices(hq)
 
         for q in qb:
             b = hq[q]
@@ -796,6 +1102,11 @@ class QrackAceBackend:
             else:
                 self.sim[b[0]].force_m(b[1], result)
 
+        b = hq[1]
+        b.reset()
+        if result:
+            b.x()
+
         return c
 
     def m_all(self):

pyqrack/qrack_simulator.py

@@ -60,6 +60,7 @@ class QrackSimulator:
         isHostPointer=(
             True if os.environ.get("PYQRACK_HOST_POINTER_DEFAULT_ON") else False
         ),
+        isSparse=False,
         noise=0,
         pyzxCircuit=None,
         qiskitCircuit=None,
@@ -99,6 +100,7 @@ class QrackSimulator:
                 isCpuGpuHybrid,
                 isOpenCL,
                 isHostPointer,
+                isSparse
             )
 
         self._throw_if_error()
@@ -1053,9 +1055,16 @@ class QrackSimulator:
         Returns:
             Measurement result of all qubits.
         """
-        result = Qrack.qrack_lib.MAll(self.sid)
+        num_q = self.num_qubits()
+        num_words = (num_q + 63) // 64
+        _r = (ctypes.c_ulonglong * num_words)()
+        Qrack.qrack_lib.MAllLong(self.sid, _r)
         self._throw_if_error()
-        return result
+        r = 0
+        for w in range(num_words):
+            r <<= 64
+            r |= _r[w]
+        return r
 
     def measure_pauli(self, b, q):
         """Pauli Measurement gate
@@ -2125,9 +2134,7 @@ class QrackSimulator:
     def decompose(self, q):
         """Decompose system
 
-        Decompose the given qubit out of the system.
-        Warning: The qubit subsystem state must be separable, or the behavior
-        of this method is undefined.
+        Factorize a set of contiguous bits with minimal fidelity loss.
 
         Args:
             q: qubit id
@@ -2137,7 +2144,7 @@ class QrackSimulator:
             RuntimeError: QrackSimulator with isTensorNetwork=True option cannot decompose()! (Turn off just this option, in the constructor.)
 
         Returns:
-            State of the systems.
+            Decomposed subsystem simulator.
         """
         if self.is_tensor_network:
             raise RuntimeError(
@@ -2154,10 +2161,8 @@ class QrackSimulator:
     def dispose(self, q):
         """Dispose qubits
 
-        Minimally decompose a set of contiguous bits from the separably
-        composed unit, and discard the separable bits.
-        Warning: The qubit subsystem state must be separable, or the behavior
-        of this method is undefined.
+        Factorize a set of contiguous bits with minimal fidelity loss,
+        and discard the separable bits.
 
         Args:
             q: qubit
@@ -2165,9 +2170,6 @@ class QrackSimulator:
         Raises:
             RuntimeError: QrackSimulator raised an exception.
             RuntimeError: QrackSimulator with isTensorNetwork=True option cannot dispose()! (Turn off just this option, in the constructor.)
-
-        Returns:
-            State of the systems.
         """
         if self.is_tensor_network:
             raise RuntimeError(
@@ -2285,6 +2287,72 @@ class QrackSimulator:
         self._throw_if_error()
         return list(probs)
 
+    def out_rdm(self, q):
+        """Get reduced density matrix
+
+        Returns the raw reduced density matrix of the simulator, for the qubit list.
+        Warning: State vector or is not always the internal representation leading
+        to sub-optimal performance of the method.
+
+        Raises:
+            RuntimeError: QrackSimulator raised an exception.
+
+        Returns:
+            flat list structure representing the reduced density matrix.
+        """
+        amp_count = 1 << len(q)
+        sqr_amp_count = amp_count * amp_count
+        flat_rdm = self._qrack_complex_byref([complex(0, 0)] * sqr_amp_count)
+        Qrack.qrack_lib.OutReducedDensityMatrix(self.sid, len(q), self._ulonglong_byref(q), flat_rdm)
+        self._throw_if_error()
+        return [complex(r, i) for r, i in self._pairwise(flat_rdm)]
+
+    def highest_prob_perm(self):
+        """Get the permutation (bit string) with the highest probability
+
+        Returns the single highest-probability bit string in the Hilbert space
+
+        Raises:
+            RuntimeError: QrackSimulator raised an exception.
+
+        Returns:
+            Highest probability dimension index
+        """
+        num_q = self.num_qubits()
+        num_words = (num_q + 63) // 64
+        _r = (ctypes.c_ulonglong * num_words)()
+        Qrack.qrack_lib.HighestProbAll(self.sid, _r)
+        self._throw_if_error()
+        r = 0
+        for w in range(num_words):
+            r <<= 64
+            r |= _r[w]
+        return r
+
+    def highest_n_prob_perm(self, n):
+        """Get the top n permutations (bit strings) with the highest probability
+
+        Returns the top n highest-probability bit strings in the Hilbert space
+
+        Raises:
+            RuntimeError: QrackSimulator raised an exception.
+
+        Returns:
+            Top n highest probability dimension indices
+        """
+        num_q = self.num_qubits()
+        num_words = (num_q + 63) // 64
+        _r = (ctypes.c_ulonglong * (num_words * n))()
+        Qrack.qrack_lib.HighestProbAllN(self.sid, n, _r)
+        self._throw_if_error()
+        r = [0] * n
+        for i in range(n):
+            r[i] = 0
+            for w in range(num_words):
+                r[i] <<= 64
+                r[i] |= _r[(i * num_words) + w]
+        return r
+
     def prob_all(self, q):
         """Probabilities of all subset permutations
 

pyqrack/qrack_system/qrack_system.py

@@ -87,19 +87,22 @@ class QrackSystem:
             CFUNCTYPE(c_ulonglong, c_double, c_double),
         ]
 
-        # These next two methods need to have c_double pointers, if PyQrack is built with fp64.
+        # These next few methods need to have c_double pointers, if PyQrack is built with fp64.
         self.qrack_lib.InKet.restype = None
         self.qrack_lib.OutKet.restype = None
         self.qrack_lib.OutProbs.restype = None
+        self.qrack_lib.OutReducedDensityMatrix.restype = None
 
         if self.fppow < 6:
             self.qrack_lib.InKet.argtypes = [c_ulonglong, POINTER(c_float)]
             self.qrack_lib.OutKet.argtypes = [c_ulonglong, POINTER(c_float)]
             self.qrack_lib.OutProbs.argtypes = [c_ulonglong, POINTER(c_float)]
+            self.qrack_lib.OutReducedDensityMatrix.argtypes = [c_ulonglong, c_ulonglong, POINTER(c_ulonglong), POINTER(c_float)]
         else:
             self.qrack_lib.InKet.argtypes = [c_ulonglong, POINTER(c_double)]
             self.qrack_lib.OutKet.argtypes = [c_ulonglong, POINTER(c_double)]
             self.qrack_lib.OutProbs.argtypes = [c_ulonglong, POINTER(c_double)]
+            self.qrack_lib.OutReducedDensityMatrix.argtypes = [c_ulonglong, c_ulonglong, POINTER(c_ulonglong), POINTER(c_double)]
 
         self.qrack_lib.init.restype = c_ulonglong
         self.qrack_lib.init.argtypes = []
@@ -108,10 +111,10 @@ class QrackSystem:
         self.qrack_lib.get_error.argtypes = [c_ulonglong]
 
         self.qrack_lib.init_count.restype = c_ulonglong
-        self.qrack_lib.init_count.argtypes = [c_ulonglong, c_bool]
+        self.qrack_lib.init_count.argtypes = [c_ulonglong, c_bool, c_bool]
 
         self.qrack_lib.init_count_pager.restype = c_ulonglong
-        self.qrack_lib.init_count_pager.argtypes = [c_ulonglong, c_bool]
+        self.qrack_lib.init_count_pager.argtypes = [c_ulonglong, c_bool, c_bool]
 
         self.qrack_lib.init_count_type.restype = c_ulonglong
         self.qrack_lib.init_count_type.argtypes = [
@@ -126,6 +129,7 @@ class QrackSystem:
             c_bool,
             c_bool,
             c_bool,
+            c_bool
         ]
 
         self.qrack_lib.init_count_stabilizer.restype = c_ulonglong
@@ -155,6 +159,12 @@ class QrackSystem:
 
         # pseudo-quantum
 
+        self.qrack_lib.HighestProbAll.restype = None
+        self.qrack_lib.HighestProbAll.argtypes = [c_ulonglong, POINTER(c_ulonglong)]
+
+        self.qrack_lib.HighestProbAllN.restype = None
+        self.qrack_lib.HighestProbAllN.argtypes = [c_ulonglong, c_ulonglong, POINTER(c_ulonglong)]
+
         self.qrack_lib.ProbAll.restype = None
         if self.fppow == 5:
             self.qrack_lib.ProbAll.argtypes = [
@@ -790,6 +800,9 @@ class QrackSystem:
         self.qrack_lib.MAll.restype = c_ulonglong
         self.qrack_lib.MAll.argtypes = [c_ulonglong]
 
+        self.qrack_lib.MAllLong.restype = None
+        self.qrack_lib.MAllLong.argtypes = [c_ulonglong, POINTER(c_ulonglong)]
+
         self.qrack_lib.Measure.restype = c_ulonglong
         self.qrack_lib.Measure.argtypes = [
             c_ulonglong,

{pyqrack_complex128-1.62.0.dist-info → pyqrack_complex128-1.71.1.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: pyqrack-complex128
-Version: 1.62.0
+Version: 1.71.1
 Summary: pyqrack - Pure Python vm6502q/qrack Wrapper
 Home-page: https://github.com/vm6502q/pyqrack
 Author: Daniel Strano
@@ -41,7 +41,7 @@ Provides-Extra: dev
 Requires-Dist: pytest>=7.3.1; extra == "dev"
 
 # pyqrack
-[![Downloads](https://pepy.tech/badge/pyqrack)](https://pepy.tech/project/pyqrack) [![Downloads](https://pepy.tech/badge/pyqrack/month)](https://pepy.tech/project/pyqrack) [![Downloads](https://static.pepy.tech/badge/pyqrack/week)](https://pepy.tech/project/pyqrack)
+[![Downloads](https://pepy.tech/badge/pyqrack-complex128)](https://pepy.tech/project/pyqrack-complex128) [![Downloads](https://pepy.tech/badge/pyqrack-complex128/month)](https://pepy.tech/project/pyqrack-complex128) [![Downloads](https://static.pepy.tech/badge/pyqrack-complex128/week)](https://pepy.tech/project/pyqrack-complex128)
 
 Pure Python bindings for the pure C++11/OpenCL Qrack quantum computer simulator library
 

{pyqrack_complex128-1.62.0.dist-info → pyqrack_complex128-1.71.1.dist-info}/RECORD

@@ -1,23 +1,23 @@
 pyqrack/__init__.py,sha256=3tBwfCCD-zQjQ2g1EUZdggKdn-3b2uSFTbT7LS0YLaU,785
 pyqrack/neuron_activation_fn.py,sha256=fQTTFfsvwcot_43Vopacot47IV2Rxk8pelUyuzwpXPs,593
 pyqrack/pauli.py,sha256=wg500wDOwdIU4lEVJoMmjtbAdmtakZYzLPjdzC2rwUQ,654
-pyqrack/qrack_ace_backend.py,sha256=SLmRwJmUAzV0BWoZJFskMYl21Kg9VXjQnoaAb4g6eDk,40334
+pyqrack/qrack_ace_backend.py,sha256=Prw1NhVVt0csbHiJeW8MJI9rl1P1YS63sXM5quoNPrI,49392
 pyqrack/qrack_circuit.py,sha256=vDCKGbcEHJDFUKprjCpWgit8lXFnMrPimKHURD2_Hj4,19538
 pyqrack/qrack_neuron.py,sha256=UiJdjAGB6usjAGHWSosSFCUUeIkhh3MtZbsaxfsIsNw,9043
 pyqrack/qrack_neuron_torch_layer.py,sha256=Bs5BLC2GFevfSpo_jSJ2AZl-hfDRJmzlGN9pFw1CtoQ,6160
-pyqrack/qrack_simulator.py,sha256=2z4zGNPvDvVz2ctjeJl3jxUnKu0CxkABHvW5zxdUHa8,143781
+pyqrack/qrack_simulator.py,sha256=PSrcEIlzCt33WXjoEo5dZdmzx9zR90QI_Dw9LTphjMg,146030
 pyqrack/qrack_stabilizer.py,sha256=O-7VJ9Vw4h25PK_kesSjIqHXGSo8lLrQLIyGgmzG7Co,2124
 pyqrack/quimb_circuit_type.py,sha256=Sk-Tmn38kUYmAkJJ75btWuhYZyTXOOezmowFhfdiGDc,621
 pyqrack/qrack_system/__init__.py,sha256=-oZ9dsb1hixsnrkUJRY_C5DzQ_l6MtifF_Z465BgqV4,334
-pyqrack/qrack_system/qrack_system.py,sha256=FWEMiBfDDoNIAo8jbqFR9MkvRGr6rOU_frvKGMgnTO8,42943
+pyqrack/qrack_system/qrack_system.py,sha256=3NFejSMePILaND-h3bdwRrX-WuhdZj-Uzt-8i90_xQE,43719
 pyqrack/qrack_system/qrack_cl_precompile/qrack_cl_precompile,sha256=8xQYqoFTKIl-5v6ZkqmbWp14Eowkl8ty2vNF2shC5hU,35040
-pyqrack/qrack_system/qrack_lib/libqrack_pinvoke.9.22.0.dylib,sha256=ZRtCsCAF67qRyTLsF9IPkEQlue-Ye0aDYIWE4L-AWFk,3136856
-pyqrack/qrack_system/qrack_lib/libqrack_pinvoke.dylib,sha256=ZRtCsCAF67qRyTLsF9IPkEQlue-Ye0aDYIWE4L-AWFk,3136856
+pyqrack/qrack_system/qrack_lib/libqrack_pinvoke.9.30.1.dylib,sha256=dFn-CqdTw3R3k08t-5zDqZRKbjMgyvSUAXk8mugjkds,3941720
+pyqrack/qrack_system/qrack_lib/libqrack_pinvoke.dylib,sha256=dFn-CqdTw3R3k08t-5zDqZRKbjMgyvSUAXk8mugjkds,3941720
 pyqrack/stats/__init__.py,sha256=Hla85my2fY_roR9lIjGBVpEG7ySOTMwjWa8D6-kgCnY,276
 pyqrack/stats/load_quantized_data.py,sha256=z12u9F7Nt3P-i44nY1xxvso_klS6WIHS3iqq7R2_lqE,1184
 pyqrack/stats/quantize_by_range.py,sha256=UM0_7jJDdQ7g30cR3UQAxkbzkqrmsy1oUfqg0h11FUY,2270
-pyqrack_complex128-1.62.0.dist-info/LICENSE,sha256=HxB-7SaWTuewAk1nz-3_3FUD6QhgX73kNT_taKVUTq8,1069
-pyqrack_complex128-1.62.0.dist-info/METADATA,sha256=lqLRMSEBTkrv5xD9kMDPrl74gVuOHbcvrrLLljwsmuY,5903
-pyqrack_complex128-1.62.0.dist-info/WHEEL,sha256=nZx8s83OrgdDmpcWX-8FgI0sEAjdQimt4SdYsdcCaC8,107
-pyqrack_complex128-1.62.0.dist-info/top_level.txt,sha256=YE_3q9JTGRLMilNg2tGP1y7uU-Dx8PDao2OhwoIbv8E,8
-pyqrack_complex128-1.62.0.dist-info/RECORD,,
+pyqrack_complex128-1.71.1.dist-info/LICENSE,sha256=HxB-7SaWTuewAk1nz-3_3FUD6QhgX73kNT_taKVUTq8,1069
+pyqrack_complex128-1.71.1.dist-info/METADATA,sha256=Xci4A4ZRU7BKKqNPv_UQmR1qyV8m8w1xl4qwnbqetUo,5969
+pyqrack_complex128-1.71.1.dist-info/WHEEL,sha256=nZx8s83OrgdDmpcWX-8FgI0sEAjdQimt4SdYsdcCaC8,107
+pyqrack_complex128-1.71.1.dist-info/top_level.txt,sha256=YE_3q9JTGRLMilNg2tGP1y7uU-Dx8PDao2OhwoIbv8E,8
+pyqrack_complex128-1.71.1.dist-info/RECORD,,