Merge branch 'multi-node-multi-GPU' of https://github.com/qiboteam/qibotn into multi-node-multi-GPU

2023-10-03 14:25:33 +08:00
parent b2a2bfedf1 fc665fcfc5
commit cec56f36d1
1 changed files with 2 additions and 1 deletions
--- a/src/qibotn/QiboCircuitConvertor.py
+++ b/src/qibotn/QiboCircuitConvertor.py
@@ -95,7 +95,8 @@ class QiboCircuitToEinsum:
            required_shape = self.op_shape_from_qubits(len(gate_qubits))
            self.gate_tensors.append(
                (
-                    cp.asarray(gate.matrix).reshape(required_shape),
+                    cp.asarray(gate.matrix(), dtype=self.dtype).reshape(
+                        required_shape),
                    gate_qubits,
                )
            )