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

Merge branch 'cuQuantum_cuTensorNet' of https://github.com/qiboteam/qibotn into cuQuantum_cuTensorNet

Browse Source
This commit is contained in:
tankya2
2023-04-21 11:11:12 +08:00
parent 9cc1dcdffb 00b6540ea6
commit dedf1dd6fd
2 changed files with 2 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
src/qibotn/QiboCircuitConvertor.py
View File

@@ -21,7 +21,7 @@ class QiboCircuitToEinsum:
"""This function is to modify the shape of the tensor to the required format by cuQuantum """This function is to modify the shape of the tensor to the required format by cuQuantum
(qubit_states,) * input_output * qubits_involved (qubit_states,) * input_output * qubits_involved
""" """
return (2,) * 2 * nqubits return (2, 2) * nqubits
self.backend = cp self.backend = cp
self.dtype = getattr(self.backend, dtype) self.dtype = getattr(self.backend, dtype)

4
src/qibotn/cutn.py
View File

@@ -5,6 +5,4 @@ from cuquantum import contract
def eval(qibo_circ, datatype): def eval(qibo_circ, datatype):
myconvertor = QiboCircuitToEinsum(qibo_circ, dtype=datatype) myconvertor = QiboCircuitToEinsum(qibo_circ, dtype=datatype)
operands_expression = myconvertor.state_vector_operands() return contract(*myconvertor.state_vector_operands())
results = contract(*operands_expression)
return results