refactor: make QibotnBackend abstract and inherit both this and NumpyBackend in Qibotn concrete backends
This commit is contained in:
@@ -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
|
||||||
|
|||||||
@@ -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."""
|
||||||
|
|
||||||
|
|||||||
@@ -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__()
|
||||||
|
|||||||
@@ -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__()
|
||||||
|
|||||||
Reference in New Issue
Block a user