Skip to content

Commit 5ddde16

Browse files
authored
Use run_batch in cirq.Sampler (#678)
1 parent 4ef56df commit 5ddde16

File tree

1 file changed

+4
-3
lines changed

1 file changed

+4
-3
lines changed

tensorflow_quantum/core/ops/cirq_ops.py

Lines changed: 4 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -529,10 +529,11 @@ def _no_grad(grad):
529529

530530
else:
531531
# All other cirq.Samplers handled here.
532-
#TODO(zaqqwerty): replace with run_batch once Cirq #3148 is resolved
533532
cirq_results = []
534-
for p, r in zip(programs, resolvers):
535-
cirq_results.append(sampler.run(p, r, num_samples))
533+
for results in sampler.run_batch(programs,
534+
params_list=resolvers,
535+
repetitions=num_samples):
536+
cirq_results.extend(results)
536537

537538
results = []
538539
for r in cirq_results:

0 commit comments

Comments
 (0)