Add datatype as an input in eval() [skip ci]

2023-04-18 11:34:00 +08:00
parent 7751dbd2f9
commit aee995802f
2 changed files with 4 additions and 3 deletions
--- a/src/qibotn/cutn.py
+++ b/src/qibotn/cutn.py
@@ -3,8 +3,8 @@ from QiboCircuitConvertor import QiboCircuitToEinsum
 from cuquantum import contract


-def eval(qibo_circ):
-    myconvertor = QiboCircuitToEinsum(qibo_circ, dtype="complex128")
+def eval(qibo_circ,datatype):
+    myconvertor = QiboCircuitToEinsum(qibo_circ, dtype=datatype)
    operands_expression = myconvertor.state_vector()
    results = contract(*operands_expression)
    return results.flatten()