Upgrading PyQrack dependency and workflows. (#504)

Apparently, in the latest version they have added a flag to disable
OpenCL, which I think will make supporting pyqrack a lot easier. I want
to try this out and see if it works!

EDIT: doesn't work, however, I can specify a fine-grained platform which
should fix the issues with which pyqrack to install.

Author: weinbe58

.github/workflows/ci.yml

@@ -24,13 +24,13 @@ jobs:
         uses: astral-sh/setup-uv@v6
         with:
           # Install a specific version of uv.
-          version: "0.5.1"
+          version: "0.8.4"
           enable-cache: true
           cache-dependency-glob: "uv.lock"
       - name: Set up Python ${{ matrix.python-version }}
         run: uv python install ${{ matrix.python-version }}
       - name: Install the project
-        run: uv sync --upgrade --dev --group dev-linux
+        run: uv sync --dev --all-extras --no-extra pyqrack-cuda
       - name: Run tests
         # For example, using `pytest`
         run: uv run just coverage

pyproject.toml

@@ -17,8 +17,8 @@ dependencies = [
     "rich>=13.9.4",
     "pydantic>=1.3.0,<2.11.0",
     "pandas>=2.2.3",
-    "pyqrack>=1.38.2,<1.41; sys_platform == 'darwin'",
-    "pyqrack-cpu>=1.38.2,<1.41 ; sys_platform != 'darwin'",
+    "pyqrack>=1.69.0; sys_platform == 'darwin-arm64'",
+    "pyqrack-cpu>=1.69.0; sys_platform != 'darwin-arm64'",
 ]
 
 [project.optional-dependencies]
@@ -41,10 +41,10 @@ cirq = [
     "qpsolvers[clarabel]>=4.7.0",
 ]
 pyqrack-opencl = [
-    "pyqrack>=1.38.2 ; sys_platform != 'darwin'",
+    "pyqrack>=1.69.0; sys_platform == 'darwin-arm64'",
 ]
 pyqrack-cuda = [
-    "pyqrack-cuda>=1.38.2",
+    "pyqrack-cuda>=1.69.0",
 ]
 stim = [
     "stim>=1.15.0",
@@ -75,34 +75,6 @@ dev = [
     "rust-just>=1.36.0",
     "tomlkit>=0.13.2",
 ]
-dev-linux = [
-    "cirq-core[contrib]>=1.4.1",
-    "lark>=1.2.2",
-    "pyqrack-cpu>=1.38.2",
-    "qbraid>=0.9.5",
-    "ffmpeg>=1.4",
-    "matplotlib>=3.9.2",
-    "pyqt5>=5.15.11",
-    "tqdm>=4.67.1",
-    "stim>=1.15.0",
-    "cirq-core>=1.4.1",
-    "cirq-core[contrib]>=1.4.1",
-    "qpsolvers[clarabel]>=4.7.0",
-]
-dev-mac-arm = [
-    "cirq-core[contrib]>=1.4.1",
-    "ffmpeg>=1.4",
-    "lark>=1.2.2",
-    "matplotlib>=3.9.2",
-    "pyqrack>=1.38.2",
-    "pyqt5>=5.15.11",
-    "qbraid>=0.9.5",
-    "tqdm>=4.67.1",
-    "stim>=1.15.0",
-    "cirq-core>=1.4.1",
-    "cirq-core[contrib]>=1.4.1",
-    "qpsolvers[clarabel]>=4.7.0",
-]
 
 [tool.isort]
 profile = "black"

test/cirq_utils/noise/test_noise_models.py

@@ -64,7 +64,9 @@ def test_simple_model(model: cirq.NoiseModel, qubits):
 
     kernel = squin.cirq.load_circuit(noisy_circuit)
     RewriteNoiseStmts(kernel.dialects)(kernel)
-    pyqrack_sim = StackMemorySimulator(min_qubits=2)
+    pyqrack_sim = StackMemorySimulator(
+        min_qubits=2, rng_state=np.random.default_rng(1234)
+    )
 
     pops_bloqade = [0.0] * 4
 

test/cirq_utils/noise/test_noisy_ghz.py

@@ -86,7 +86,9 @@ def fidelity(rho: np.ndarray, sigma: np.ndarray) -> float:
             # do the same in squin
             kernel = squin.cirq.load_circuit(noisy_circuit)
             RewriteNoiseStmts(kernel.dialects)(kernel)
-            sim = StackMemorySimulator(min_qubits=n)
+            sim = StackMemorySimulator(
+                min_qubits=n, rng_state=np.random.default_rng(1234)
+            )
             rho_squin = np.zeros((2**n, 2**n), dtype=np.complex128)
             nshots = 300
             for _ in range(nshots):

test/cirq_utils/noise/test_one_zone_correlated_noise.py

@@ -51,7 +51,7 @@ def test_model_with_defaults():
 
     RewriteNoiseStmts(kernel.dialects)(kernel)
 
-    sim = StackMemorySimulator(min_qubits=2)
+    sim = StackMemorySimulator(min_qubits=2, rng_state=np.random.default_rng(1234))
     pops = [0.0] * 4
     nshots = 300
     for _ in range(nshots):

test/pyqrack/squin/test_kernel.py

@@ -1,5 +1,6 @@
 import math
 
+import numpy as np
 import pytest
 from kirin.dialects import ilist
 
@@ -22,7 +23,15 @@ def new():
     assert isinstance(qubit := result[0], PyQrackQubit)
 
     out = qubit.sim_reg.out_ket()
-    assert out == [1.0] + [0.0] * (2**3 - 1)
+    out = np.asarray(out)
+
+    i = np.abs(out).argmax()
+    out /= out[i] / np.abs(out[i])
+
+    expected = np.zeros_like(out)
+    expected[0] = 1.0
+
+    assert np.allclose(out, expected, atol=2.2e-7)
 
     @squin.kernel
     def m():

test/pyqrack/test_target.py

@@ -1,6 +1,7 @@
 import math
 import textwrap
 
+import numpy as np
 import pytest
 from kirin import ir
 from kirin.dialects import ilist
@@ -115,11 +116,14 @@ def multiple_registers():
 
     out = qubit.sim_reg.out_ket()
 
-    assert out[0] == 1
-    for i in range(1, len(out)):
-        assert out[i] == 0
+    out = np.asarray(out)
+    i = np.abs(out).argmax()
+    out /= out[i] / np.abs(out[i])
 
-    assert True
+    expected = np.zeros_like(out)
+    expected[0] = 1.0
+
+    assert np.allclose(out, expected, atol=2.2e-7)
 
 
 def test_measurement():