refactor: make QibotnBackend abstract and inherit both this and NumpyBackend in Qibotn concrete backends

This commit is contained in:
MatteoRobbiati
2025-02-10 10:49:30 +01:00
parent d854d15068
commit a5f7d1fb14
4 changed files with 10 additions and 17 deletions

View File

@@ -1,20 +1,11 @@
from qibo.backends.numpy import NumpyBackend from abc import ABC
from qibo.config import raise_error from qibo.config import raise_error
DEFAULT_CONFIGURATION = {
"MPI_enabled": False, # TODO: cutensornet specific, TBRemoved
"NCCL_enabled": False, # TODO: cutensornet specific, TBRemoved
"expectation_enabled": False,
"pauli_string_pattern": None,
"MPS_enabled": False,
"gate_algo": None,
"mps_opts": None,
}
class QibotnBackend(ABC):
class QibotnBackend(NumpyBackend): def __init__(self):
def __init__(self, runcard: dict = DEFAULT_CONFIGURATION):
super().__init__() super().__init__()
def apply_gate(self, gate, state, nqubits): # pragma: no cover def apply_gate(self, gate, state, nqubits): # pragma: no cover
@@ -33,7 +24,6 @@ class QibotnBackend(NumpyBackend):
def set_device(self, device): def set_device(self, device):
self.device = device self.device = device
# @abstractmethod
def configure_tn_simulation(self, **config): def configure_tn_simulation(self, **config):
"""Configure the TN simulation that will be performed.""" """Configure the TN simulation that will be performed."""
pass pass

View File

@@ -1,4 +1,5 @@
import numpy as np import numpy as np
from qibo.backends import NumpyBackend
from qibo.config import raise_error from qibo.config import raise_error
from qibo.result import QuantumState from qibo.result import QuantumState
@@ -7,7 +8,7 @@ from qibotn.backends.abstract import QibotnBackend
CUDA_TYPES = {} CUDA_TYPES = {}
class CuTensorNet(QibotnBackend): # pragma: no cover class CuTensorNet(NumpyBackend, QibotnBackend): # pragma: no cover
# CI does not test for GPU # CI does not test for GPU
"""Creates CuQuantum backend for QiboTN.""" """Creates CuQuantum backend for QiboTN."""

View File

@@ -7,6 +7,7 @@ import numpy as np
import qiskit import qiskit
import qmatchatea import qmatchatea
import qtealeaves import qtealeaves
from qibo.backends import NumpyBackend
from qibo.config import raise_error from qibo.config import raise_error
from qibotn.backends.abstract import QibotnBackend from qibotn.backends.abstract import QibotnBackend
@@ -14,7 +15,7 @@ from qibotn.result import TensorNetworkResult
@dataclass @dataclass
class QMatchaTeaBackend(QibotnBackend): class QMatchaTeaBackend(NumpyBackend, QibotnBackend):
def __init__(self): def __init__(self):
super().__init__() super().__init__()

View File

@@ -1,10 +1,11 @@
from qibo.backends import NumpyBackend
from qibo.config import raise_error from qibo.config import raise_error
from qibo.result import QuantumState from qibo.result import QuantumState
from qibotn.backends.abstract import QibotnBackend from qibotn.backends.abstract import QibotnBackend
class QuimbBackend(QibotnBackend): class QuimbBackend(NumpyBackend, QibotnBackend):
def __init__(self, runcard): def __init__(self, runcard):
super().__init__() super().__init__()