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

Change output to QuantumState class

Browse Source
This commit is contained in:
tankya2
2024-02-22 17:19:39 +08:00
parent 20589760f1
commit 52da762e1b
1 changed files with 3 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
src/qibotn/backends/cutensornet.py
View File

@@ -2,7 +2,7 @@ import cuquantum # pylint: disable=import-error
import numpy as np import numpy as np
from qibo.backends.numpy import NumpyBackend from qibo.backends.numpy import NumpyBackend
from qibo.config import raise_error from qibo.config import raise_error
from qibo.result import CircuitResult from qibo.result import QuantumState
CUDA_TYPES = { CUDA_TYPES = {
"complex64": ( "complex64": (
@@ -185,5 +185,5 @@ class CuTensorNet(NumpyBackend): # pragma: no cover
if return_array: if return_array:
return state.flatten() return state.flatten()
else: else:
circuit._final_state = CircuitResult(self, circuit, state.flatten(), nshots) return QuantumState(state.flatten())
return circuit._final_state