jaunatisblue/final-qibotn
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

merge main

Browse Source
This commit is contained in:
BrunoLiegiBastonLiegi
2025-12-17 09:45:18 +01:00
parent 828c86c0a8 18a5371a30
commit 4137177c03
7 changed files with 1116 additions and 567 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
.pre-commit-config.yaml
View File

@@ -10,7 +10,7 @@ repos:
- id: check-toml - id: check-toml
- id: debug-statements - id: debug-statements
- repo: https://github.com/psf/black-pre-commit-mirror - repo: https://github.com/psf/black-pre-commit-mirror
rev: 25.11.0 rev: 25.12.0
hooks: hooks:
- id: black - id: black
- repo: https://github.com/pycqa/isort - repo: https://github.com/pycqa/isort

1658
poetry.lock generated
View File

File diff suppressed because it is too large Load Diff

11
pyproject.toml
View File

@@ -22,17 +22,20 @@ packages = [{ include = "qibotn", from = "src" }]
[tool.poetry.dependencies] [tool.poetry.dependencies]
python = ">=3.11,<3.14" python = ">=3.11,<3.14"
qibo = { git="https://github.com/qiboteam/qibo", branch="backends" } qibo = { git="https://github.com/qiboteam/qibo", branch="backends" }
qibojit = { git = "https://github.com/qiboteam/qibojit.git", branch="backends" }
quimb = { version = "^1.10.0", extras = ["tensor"] } quimb = { version = "^1.10.0", extras = ["tensor"] }
cupy-cuda11x = { version = "^13.1.0", optional = true } cupy-cuda12x = { version = "^13.6.0", optional = true }
cuquantum-python-cu11 = { version = "^24.1.0", optional = true } cuda-toolkit = {extras = ["all"], version = "^12.9.1"}
nvidia-nccl-cu12 = { version = "^2.16.5", optional = true }
cuquantum-python-cu12 = { version = "^25.9.1", optional = true }
qmatchatea = { version = "^1.4.3", optional = true } qmatchatea = { version = "^1.4.3", optional = true }
qiskit = { version = "^1.4.0", optional = true } qiskit = { version = "^1.4.0", optional = true }
qtealeaves = { version = "^1.5.20", optional = true } qtealeaves = { version = "^1.5.20", optional = true }
[tool.poetry.extras] [tool.poetry.extras]
cuda = ["cupy-cuda11x", "cuquantum-python-cu11", "mpi4py"] cuda = ["cupy-cuda12x", "cuda-toolkit", "nvidia-nccl-cu12", "cuquantum-python-cu12", "mpi4py"]
qmatchatea = ["qmatchatea", "qtealeaves", "qiskit"] qmatchatea = ["qmatchatea"]
[tool.poetry.group.docs] [tool.poetry.group.docs]
optional = true optional = true

2
src/qibotn/circuit_to_mps.py
View File

@@ -1,6 +1,6 @@
import cupy as cp import cupy as cp
import cuquantum.bindings.cutensornet as cutn
import numpy as np import numpy as np
from cuquantum import cutensornet as cutn
from qibotn.circuit_convertor import QiboCircuitToEinsum from qibotn.circuit_convertor import QiboCircuitToEinsum
from qibotn.mps_utils import apply_gate, initial from qibotn.mps_utils import apply_gate, initial

4
src/qibotn/eval.py
View File

@@ -1,8 +1,8 @@
import cupy as cp import cupy as cp
import cuquantum.cutensornet as cutn import cuquantum.bindings.cutensornet as cutn
from cupy.cuda import nccl from cupy.cuda import nccl
from cupy.cuda.runtime import getDeviceCount from cupy.cuda.runtime import getDeviceCount
from cuquantum import Network, contract from cuquantum.tensornet import Network, contract
from mpi4py import MPI from mpi4py import MPI
from qibo import hamiltonians from qibo import hamiltonians
from qibo.symbols import I, X, Y, Z from qibo.symbols import I, X, Y, Z

2
src/qibotn/mps_contraction_helper.py
View File

@@ -1,4 +1,4 @@
from cuquantum import contract, contract_path from cuquantum.tensornet import contract, contract_path
# Reference: https://github.com/NVIDIA/cuQuantum/blob/main/python/samples/cutensornet/tn_algorithms/mps_algorithms.ipynb # Reference: https://github.com/NVIDIA/cuQuantum/blob/main/python/samples/cutensornet/tn_algorithms/mps_algorithms.ipynb

4
src/qibotn/mps_utils.py
View File

@@ -1,6 +1,6 @@
import cupy as cp import cupy as cp
from cuquantum import contract from cuquantum.tensornet import contract
from cuquantum.cutensornet.experimental import contract_decompose from cuquantum.tensornet.experimental import contract_decompose
def initial(num_qubits, dtype): def initial(num_qubits, dtype):