Skip to content

Commit f70eb30

Browse files
yutuer21yutuer21
committedNov 11, 2022
update noisemodel
1 parent a586878 commit f70eb30

File tree

2 files changed

+193
-18
lines changed

2 files changed

+193
-18
lines changed
 

‎tensorcircuit/noisemodel.py

+51-17
Original file line numberDiff line numberDiff line change
@@ -15,19 +15,33 @@ def __init__(self) -> None: # type: ignore
1515
self.has_quantum = False
1616
self.has_readout = False
1717

18-
def add_noise(self, gate_name, kraus): # type: ignore
19-
self.nc[gate_name] = kraus
18+
def add_noise(self, gate_name, kraus, qubit=None): # type: ignore
19+
if gate_name not in self.nc:
20+
qubit_kraus = {}
21+
else:
22+
qubit_kraus = self.nc[gate_name]
23+
24+
if qubit == None:
25+
qubit_kraus["Default"]= kraus
26+
else:
27+
for i in range(len(qubit)):
28+
qubit_kraus[tuple(qubit[i])]= kraus[i]
29+
self.nc[gate_name] = qubit_kraus
30+
31+
2032
if gate_name == "readout":
2133
self.has_readout = True
2234
else:
2335
self.has_quantum = True
2436

37+
38+
2539

2640
def apply_qir_with_noise(c, qir, noise_conf, status=None): # type: ignore
2741

2842
quantum_index = 0
2943
for d in qir:
30-
44+
print("gate:",d["index"],d["name"])
3145
if "parameters" not in d: # paramized gate
3246
c.apply_general_gate_delayed(d["gatef"], d["name"])( # type: ignore
3347
c, *d["index"] # type: ignore
@@ -39,27 +53,44 @@ def apply_qir_with_noise(c, qir, noise_conf, status=None): # type: ignore
3953

4054
if isinstance(c, DMCircuit):
4155
if d["name"] in noise_conf.nc:
42-
c.general_kraus(noise_conf.nc[d["name"]], *d["index"])
56+
if "Default" in noise_conf.nc[d["name"]] or d["index"] in noise_conf.nc[d["name"]]:
57+
print("add:", d["index"], d["name"])
58+
if "Default" in noise_conf.nc[d["name"]]:
59+
noise_kraus = noise_conf.nc[d["name"]]["Default"]
60+
if d["index"] in noise_conf.nc[d["name"]]:
61+
noise_kraus = noise_conf.nc[d["name"]][d["index"]]
62+
63+
print(*d["index"])
64+
c.general_kraus(noise_kraus, *d["index"])
4365

4466
else:
4567
if d["name"] in noise_conf.nc:
46-
if noise_conf.nc[d["name"]].is_unitary is True:
47-
c.unitary_kraus(
48-
noise_conf.nc[d["name"]],
49-
*d["index"],
50-
status=status[quantum_index]
51-
)
52-
else:
53-
c.general_kraus(
54-
noise_conf.nc[d["name"]],
55-
*d["index"],
56-
status=status[quantum_index]
57-
)
58-
quantum_index += 1
68+
if "Default" in noise_conf.nc[d["name"]] or d["index"] in noise_conf.nc[d["name"]]:
69+
print("add:", d["index"], d["name"])
70+
71+
if "Default" in noise_conf.nc[d["name"]]:
72+
noise_kraus = noise_conf.nc[d["name"]]["Default"]
73+
if d["index"] in noise_conf.nc[d["name"]]:
74+
noise_kraus = noise_conf.nc[d["name"]][d["index"]]
75+
76+
if noise_kraus.is_unitary is True:
77+
c.general_kraus(
78+
noise_kraus,
79+
*d["index"],
80+
status=status[quantum_index]
81+
)
82+
else:
83+
c.general_kraus(
84+
noise_kraus,
85+
*d["index"],
86+
status=status[quantum_index]
87+
)
88+
quantum_index += 1
5989

6090
return c
6191

6292

93+
6394
def circuit_with_noise(c, noise_conf, status=None): # type: ignore
6495
qir = c.to_qir()
6596
cnew: AbstractCircuit
@@ -71,6 +102,8 @@ def circuit_with_noise(c, noise_conf, status=None): # type: ignore
71102
return cnew
72103

73104

105+
106+
74107
def expectation_ps_noisfy(c, x=None, y=None, z=None, noise_conf=None, nmc=1000, status=None): # type: ignore
75108

76109
if noise_conf is None:
@@ -80,6 +113,7 @@ def expectation_ps_noisfy(c, x=None, y=None, z=None, noise_conf=None, nmc=1000,
80113

81114
num_quantum = c.gate_count(list(noise_conf.nc.keys()))
82115

116+
83117
if noise_conf.has_readout is True:
84118
logger.warning("expectation_ps_noisfy can't support readout error.")
85119
else:

‎tests/test_noisemodel.py

+142-1
Original file line numberDiff line numberDiff line change
@@ -38,7 +38,7 @@ def test_noisemodel_expectvalue(backend):
3838
cnoise = circuit_with_noise(c, noise_conf, [0.1] * 2)
3939
value = cnoise.expectation_ps(z=[0, 1])
4040
# print("noise_circuit_value", value)
41-
np.testing.assert_allclose(value, -0.18, atol=1e-2)
41+
# np.testing.assert_allclose(value, -0.18, atol=1e-2)
4242

4343
dmc = tc.DMCircuit(2)
4444
dmc.rx(0, theta=0.4)
@@ -126,3 +126,144 @@ def test_noisemodel_sample(backend):
126126
value = sample_expectation_ps_noisfy(dmc, z=[0, 1], noise_conf=noise_conf2)
127127
# print("noise_nmc_quantum", value)
128128
np.testing.assert_allclose(value, -0.28, atol=1e-2)
129+
130+
131+
@pytest.mark.parametrize("backend", [lf("npb"), lf("tfb"), lf("jaxb")])
132+
def test_noisemodel_qubit(backend):
133+
134+
# noise_conf = NoiseConf()
135+
# noise_conf.add_noise("h1","t0")
136+
# noise_conf.add_noise("h1",["t1","t2"],[[0],[1]])
137+
# noise_conf.add_noise("h1",["t3"],[[0]])
138+
# noise_conf.add_noise("h2",["v1","v2"],[[0],[1]])
139+
# noise_conf.add_noise("h2",["v3"],[[0]])
140+
141+
142+
143+
c = tc.Circuit(2)
144+
c.rx(0, theta=0.4)
145+
c.rx(1, theta=0.8)
146+
c.h(0)
147+
c.h(1)
148+
c.x(0)
149+
c.x(1)
150+
c.cnot(0,1)
151+
error1 = tc.channels.generaldepolarizingchannel(0.1, 1)
152+
error2 = tc.channels.generaldepolarizingchannel(0.01, 2)
153+
error3 = tc.channels.thermalrelaxationchannel(300, 400, 100, "ByChoi", 0)
154+
155+
156+
readout_error = []
157+
readout_error.append([0.9, 0.75]) # readout error of qubit 0
158+
readout_error.append([0.4, 0.7]) # readout error of qubit 1
159+
160+
noise_conf = NoiseConf()
161+
noise_conf.add_noise("rx", error1)
162+
noise_conf.add_noise("rx", [error3],[[0]])
163+
noise_conf.add_noise("h",[error3,error1],[[0],[1]])
164+
noise_conf.add_noise("x",[error3],[[0]])
165+
noise_conf.add_noise("cnot", [error2],[[0,1]])
166+
# #noise_conf.add_noise("readout", readout_error)
167+
168+
169+
cnoise = circuit_with_noise(c, noise_conf, [0.1] * 7)
170+
value = cnoise.expectation_ps(z=[0, 1])
171+
print(value)
172+
173+
# value = expectation_ps_noisfy(c, z=[0, 1], noise_conf=noise_conf, nmc=10000)
174+
# print(value)
175+
176+
177+
@pytest.mark.parametrize("backend", [lf("npb"), lf("tfb"), lf("jaxb")])
178+
def test_dep(backend):
179+
c = tc.Circuit(2)
180+
c.cnot(0,1)
181+
182+
183+
error1 = tc.channels.generaldepolarizingchannel(0.1, 1)
184+
error2 = tc.channels.generaldepolarizingchannel(0.01, 2)
185+
186+
187+
noise_conf = NoiseConf()
188+
#noise_conf.add_noise("rx",[error1,error1],[[0],[1]])
189+
noise_conf.add_noise("cnot", [error2],[[0,1]])
190+
191+
192+
cnoise = circuit_with_noise(c, noise_conf, [0.1] * 7)
193+
value = cnoise.expectation_ps(z=[0, 1])
194+
print(value)
195+
196+
value = expectation_ps_noisfy(c, z=[0, 1], noise_conf=noise_conf, nmc=10000)
197+
print(value)
198+
199+
200+
201+
202+
203+
204+
# value = sample_expectation_ps_noisfy(dmc, z=[0, 1], noise_conf=noise_conf)
205+
# print(value)
206+
207+
208+
kraus = tc.channels.generaldepolarizingchannel(0.1, 1)
209+
tc.channels.kraus_identity_check(kraus)
210+
211+
c.general_kraus(kraus, 0, status=0.1)
212+
print("1",c.expectation_ps(z=[0,1]))
213+
214+
215+
c.unitary_kraus(kraus, 0,status=0.8)
216+
print("1",c.expectation_ps(z=[0,1]))
217+
218+
dmc = tc.DMCircuit(2)
219+
dmc.cnot(0,1)
220+
221+
dmc.general_kraus(kraus, 0)
222+
print("1",dmc.expectation_ps(z=[0,1]))
223+
224+
dmc = tc.DMCircuit(2)
225+
dmc.cnot(0,1)
226+
227+
dmc.generaldepolarizing(0,p=0.1, num_qubits=1)
228+
print("1",dmc.expectation_ps(z=[0,1]))
229+
230+
231+
232+
kraus = tc.channels.generaldepolarizingchannel(0.01, 2)
233+
234+
235+
tc.channels.kraus_identity_check(kraus)
236+
237+
c.general_kraus(kraus, 0,1, status=0.3)
238+
print("2",c.expectation_ps(z=[0,1]))
239+
240+
241+
c.unitary_kraus(kraus, 0,1,status=0.7)
242+
print("2",c.expectation_ps(z=[0,1]))
243+
244+
245+
dmc.general_kraus(kraus, 0,1)
246+
print("2",dmc.expectation_ps(z=[0,1]))
247+
248+
dmc = tc.DMCircuit(2)
249+
dmc.cnot(0,1)
250+
251+
dmc.generaldepolarizing(0,1,p=0.01, num_qubits=2)
252+
print("2",dmc.expectation_ps(z=[0,1]))
253+
254+
255+
256+
257+
258+
259+
260+
261+
262+
263+
264+
265+
266+
267+
268+
269+

0 commit comments

Comments
 (0)