Remove backend
This commit is contained in:
tankya2
@@ -1,129 +0,0 @@
|
|||||||
from qibo.backends import NumpyBackend
|
|
||||||
from qibo.config import raise_error
|
|
||||||
from qibotn import cutn
|
|
||||||
from qibotn import quimb
|
|
||||||
from qibo.states import CircuitResult
|
|
||||||
import numpy as np
|
|
||||||
|
|
||||||
|
|
||||||
class QiboTNBackend(NumpyBackend):
|
|
||||||
def __init__(self, platform):
|
|
||||||
super().__init__()
|
|
||||||
self.name = "qibotn"
|
|
||||||
if (
|
|
||||||
platform == "cu_tensornet"
|
|
||||||
or platform == "cu_mps"
|
|
||||||
or platform == "qu_tensornet"
|
|
||||||
or platform == "cu_tensornet_mpi"
|
|
||||||
or platform == "cu_tensornet_mpi_expectation"
|
|
||||||
or platform == "cu_tensornet_expectation"
|
|
||||||
or platform == "cu_tensornet_nccl"
|
|
||||||
or platform == "cu_tensornet_nccl_expectation"
|
|
||||||
): # pragma: no cover
|
|
||||||
self.platform = platform
|
|
||||||
else:
|
|
||||||
raise_error(
|
|
||||||
NotImplementedError, "QiboTN cannot support the specified backend."
|
|
||||||
)
|
|
||||||
|
|
||||||
def apply_gate(self, gate, state, nqubits): # pragma: no cover
|
|
||||||
raise_error(NotImplementedError, "QiboTN cannot apply gates directly.")
|
|
||||||
|
|
||||||
def apply_gate_density_matrix(self, gate, state, nqubits): # pragma: no cover
|
|
||||||
raise_error(NotImplementedError, "QiboTN cannot apply gates directly.")
|
|
||||||
|
|
||||||
def assign_measurements(self, measurement_map, circuit_result):
|
|
||||||
raise_error(NotImplementedError, "Not implemented in QiboTN.")
|
|
||||||
|
|
||||||
def execute_circuit(
|
|
||||||
self, circuit, initial_state=None, nshots=None, return_array=False
|
|
||||||
): # pragma: no cover
|
|
||||||
"""Executes a quantum circuit.
|
|
||||||
|
|
||||||
Args:
|
|
||||||
circuit (:class:`qibo.models.circuit.Circuit`): Circuit to execute.
|
|
||||||
initial_state (:class:`qibo.models.circuit.Circuit`): Circuit to prepare the initial state.
|
|
||||||
If ``None`` the default ``|00...0>`` state is used.
|
|
||||||
|
|
||||||
Returns:
|
|
||||||
xxx.
|
|
||||||
|
|
||||||
"""
|
|
||||||
|
|
||||||
if self.platform == "cu_tensornet":
|
|
||||||
if initial_state is not None:
|
|
||||||
raise_error(NotImplementedError, "QiboTN cannot support initial state.")
|
|
||||||
|
|
||||||
state = cutn.eval(circuit, self.dtype)
|
|
||||||
|
|
||||||
if self.platform == "cu_mps":
|
|
||||||
if initial_state is not None:
|
|
||||||
raise_error(NotImplementedError, "QiboTN cannot support initial state.")
|
|
||||||
|
|
||||||
gate_algo = {
|
|
||||||
"qr_method": False,
|
|
||||||
"svd_method": {
|
|
||||||
"partition": "UV",
|
|
||||||
"abs_cutoff": 1e-12,
|
|
||||||
},
|
|
||||||
} # make this user input
|
|
||||||
state = cutn.eval_mps(circuit, gate_algo, self.dtype)
|
|
||||||
|
|
||||||
if self.platform == "qu_tensornet":
|
|
||||||
# init_state = np.random.random(2**circuit.nqubits) + 1j * np.random.random(2**circuit.nqubits)
|
|
||||||
# init_state = init_state / np.sqrt((np.abs(init_state) ** 2).sum())
|
|
||||||
init_state = np.zeros(2**circuit.nqubits, dtype=self.dtype)
|
|
||||||
init_state[0] = 1.0
|
|
||||||
state = quimb.eval(circuit.to_qasm(), init_state, backend="numpy")
|
|
||||||
|
|
||||||
if self.platform == "cu_tensornet_mpi":
|
|
||||||
if initial_state is not None:
|
|
||||||
raise_error(NotImplementedError, "QiboTN cannot support initial state.")
|
|
||||||
|
|
||||||
# state, rank = cutn.eval_tn_MPI(circuit, self.dtype,32)
|
|
||||||
state, rank = cutn.eval_tn_MPI_2(circuit, self.dtype, 32)
|
|
||||||
if rank > 0:
|
|
||||||
state = np.array(0)
|
|
||||||
|
|
||||||
if self.platform == "cu_tensornet_nccl":
|
|
||||||
if initial_state is not None:
|
|
||||||
raise_error(NotImplementedError, "QiboTN cannot support initial state.")
|
|
||||||
|
|
||||||
# state, rank = cutn.eval_tn_MPI(circuit, self.dtype,32)
|
|
||||||
state, rank = cutn.eval_tn_nccl(circuit, self.dtype, 32)
|
|
||||||
if rank > 0:
|
|
||||||
state = np.array(0)
|
|
||||||
|
|
||||||
if self.platform == "cu_tensornet_expectation":
|
|
||||||
if initial_state is not None:
|
|
||||||
raise_error(NotImplementedError, "QiboTN cannot support initial state.")
|
|
||||||
|
|
||||||
state = cutn.eval_expectation(circuit, self.dtype)
|
|
||||||
|
|
||||||
if self.platform == "cu_tensornet_mpi_expectation":
|
|
||||||
if initial_state is not None:
|
|
||||||
raise_error(NotImplementedError, "QiboTN cannot support initial state.")
|
|
||||||
|
|
||||||
# state, rank = cutn.eval_tn_MPI(circuit, self.dtype,32)
|
|
||||||
# state, rank = cutn.eval_tn_MPI_expectation(circuit, self.dtype,32)
|
|
||||||
state, rank = cutn.eval_tn_MPI_2_expectation(circuit, self.dtype, 32)
|
|
||||||
|
|
||||||
if rank > 0:
|
|
||||||
state = np.array(0)
|
|
||||||
|
|
||||||
if self.platform == "cu_tensornet_nccl_expectation":
|
|
||||||
if initial_state is not None:
|
|
||||||
raise_error(NotImplementedError, "QiboTN cannot support initial state.")
|
|
||||||
|
|
||||||
# state, rank = cutn.eval_tn_MPI(circuit, self.dtype,32)
|
|
||||||
# state, rank = cutn.eval_tn_MPI_expectation(circuit, self.dtype,32)
|
|
||||||
state, rank = cutn.eval_tn_nccl_expectation(circuit, self.dtype, 32)
|
|
||||||
|
|
||||||
if rank > 0:
|
|
||||||
state = np.array(0)
|
|
||||||
|
|
||||||
if return_array:
|
|
||||||
return state.flatten()
|
|
||||||
else:
|
|
||||||
circuit._final_state = CircuitResult(self, circuit, state.flatten(), nshots)
|
|
||||||
return circuit._final_state
|
|
||||||
Reference in New Issue
Block a user