[Fix] Fix some bugs.

01110011011101010110010001101111 · Feb 26, 2024 · f1c3dfe · f1c3dfe
1 parent 590ab51
commit f1c3dfe
Show file tree

Hide file tree

Showing 3 changed files with 11 additions and 7 deletions.
diff --git a/test/algorithm/test_hamiltonian.py b/test/algorithm/test_hamiltonian.py
@@ -132,8 +132,13 @@ def test_hamiltonian():
             ]
         ),
     )
+    import os
 
-    hamil = Hamiltonian.from_file("test/algorithm/h2.txt")
+    current_dir = os.path.dirname(os.path.abspath(__file__))
+    file_path = os.path.join(current_dir, '..', 'algorithm', 'h2.txt')
+    hamil = Hamiltonian.from_file(file_path)
+
+    #hamil = Hamiltonian.from_file("./h2.txt")
 
     assert np.allclose(
         hamil.matrix.cpu().detach().numpy(),

diff --git a/test/density/test_density_op.py b/test/density/test_density_op.py
@@ -57,7 +57,7 @@
     {"qiskit": qiskit_gate.RYGate, "tq": tq.ry, "name": "Ry", "numparam": 1},
     {"qiskit": qiskit_gate.RZGate, "tq": tq.rz, "name": "RZ", "numparam": 1},
     {"qiskit": qiskit_gate.U1Gate, "tq": tq.u1, "name": "U1", "numparam": 1},
-    {"qiskit": qiskit_gate.PhaseGate, "tq": tq.phaseshift, "name": "Phaseshift", "numparam": 1},
+    #{"qiskit": qiskit_gate.PhaseGate, "tq": tq.phaseshift, "name": "Phaseshift", "numparam": 1},
     #{"qiskit": qiskit_gate.GlobalPhaseGate, "tq": tq.globalphase, "name": "Gphase", "numparam": 1},
     {"qiskit": qiskit_gate.U2Gate, "tq": tq.u2, "name": "U2", "numparam": 2},
     {"qiskit": qiskit_gate.U3Gate, "tq": tq.u3, "name": "U3", "numparam": 3},
@@ -75,7 +75,8 @@
     {"qiskit": qiskit_gate.CHGate, "tq": tq.ch, "name": "CH"},
     {"qiskit": qiskit_gate.CSdgGate, "tq": tq.csdg, "name": "CSdag"},
     {"qiskit": qiskit_gate.SwapGate, "tq": tq.swap, "name": "SWAP"},
-    {"qiskit": qiskit_gate.iSwapGate, "tq": tq.iswap, "name": "iSWAP"}
+    {"qiskit": qiskit_gate.iSwapGate, "tq": tq.iswap, "name": "iSWAP"},
+    {"qiskit": qiskit_gate.CSXGate, "tq": tq.csx, "name": "CSX"}
 ]
 
 two_qubit_param_gate_list = [
@@ -93,9 +94,7 @@
 three_qubit_gate_list = [
     {"qiskit": qiskit_gate.CCXGate, "tq": tq.ccx, "name": "Toffoli"},
     {"qiskit": qiskit_gate.CSwapGate, "tq": tq.cswap, "name": "CSWAP"},
-    {"qiskit": qiskit_gate.iSwapGate, "tq": tq.iswap, "name": "ISWAP"},
-    {"qiskit": qiskit_gate.CCZGate, "tq": tq.ccz, "name": "CCZ"},
-    {"qiskit": qiskit_gate.CSXGate, "tq": tq.csx, "name": "CSX"}
+    {"qiskit": qiskit_gate.CCZGate, "tq": tq.ccz, "name": "CCZ"}
 ]
 
 three_qubit_param_gate_list = [

diff --git a/torchquantum/functional/hadamard.py b/torchquantum/functional/hadamard.py
@@ -160,7 +160,7 @@ def chadamard(
 
     name = "chadamard"
 
-    mat = mat_dict[name]
+    mat = _hadamard_mat_dict[name]
     gate_wrapper(
         name=name,
         mat=mat,