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

Update src/qibotn/backends/cpu.py

Browse Source 
Co-authored-by: Alessandro Candido <candido.ale@gmail.com>
This commit is contained in:
Vinitha-balachandran
2024-02-08 17:58:53 +08:00
committed by GitHub
parent e58d176de4
commit a2e32c8873
1 changed files with 6 additions and 66 deletions
Download Patch File Download Diff File Expand all files Collapse all files

72
src/qibotn/backends/cpu.py
View File

@@ -62,80 +62,20 @@ class QuimbBackend(NumpyBackend):
import qibotn.eval_qu as eval
if (
self.MPI_enabled == False
and self.MPS_enabled == False
and self.NCCL_enabled == False
and self.expectation_enabled == False
):
state = eval.dense_vector_tn_qu(
circuit.to_qasm(), initial_state, is_mps=False, backend="numpy"
)
elif (
self.MPI_enabled == False
and self.MPS_enabled == True
and self.NCCL_enabled == False
and self.expectation_enabled == False
):
state = eval.dense_vector_tn_qu(
circuit.to_qasm(), initial_state, is_mps=True, backend="numpy"
)
elif (
self.MPI_enabled == True
and self.MPS_enabled == False
and self.NCCL_enabled == False
and self.expectation_enabled == False
):
if self.MPI_enabled == True:
raise_error(NotImplementedError, "QiboTN quimb backend cannot support MPI.")
elif (
self.MPI_enabled == False
and self.MPS_enabled == False
and self.NCCL_enabled == True
and self.expectation_enabled == False
):
if self.NCCL_enabled == True:
raise_error(
NotImplementedError, "QiboTN quimb backend cannot support NCCL."
)
elif (
self.MPI_enabled == False
and self.MPS_enabled == False
and self.NCCL_enabled == False
and self.expectation_enabled == True
):
if self.expectation_enabled == True:
raise_error(
NotImplementedError, "QiboTN quimb backend cannot support expectation"
)
elif (
self.MPI_enabled == True
and self.MPS_enabled == False
and self.NCCL_enabled == False
and self.expectation_enabled == True
):
raise_error(
NotImplementedError, "QiboTN quimb backend cannot support expectation"
)
elif (
self.MPI_enabled == False
and self.MPS_enabled == False
and self.NCCL_enabled == True
and self.expectation_enabled == True
):
raise_error(
NotImplementedError, "QiboTN quimb backend cannot support expectation"
)
else:
raise_error(NotImplementedError, "Compute type not supported.")
state = eval.dense_vector_tn_qu(
circuit.to_qasm(), initial_state, is_mps=self.MPS_enabled, backend="numpy"
)
if return_array:
return state.flatten()