Skip to content

Commit 614c78a

Browse files
authored
Set maxsize of request queue to Quantum Engine (#6576)
* set maxsize for request queue * add max message size
1 parent 37e8df9 commit 614c78a

File tree

3 files changed

+7
-5
lines changed

3 files changed

+7
-5
lines changed

cirq-google/cirq_google/cloud/quantum_v1alpha1/services/quantum_engine_service/transports/grpc.py

+3-2
Original file line numberDiff line numberDiff line change
@@ -169,8 +169,9 @@ def __init__(
169169
ssl_credentials=self._ssl_channel_credentials,
170170
quota_project_id=quota_project_id,
171171
options=[
172-
("grpc.max_send_message_length", -1),
173-
("grpc.max_receive_message_length", -1),
172+
('grpc.max_send_message_length', 20 * 1024 * 1024), # 20MiB
173+
('grpc.max_receive_message_length', -1), # unlimited
174+
('grpc.max_metadata_length', 10 * 1024 * 1024), # 10MiB
174175
],
175176
)
176177

cirq-google/cirq_google/cloud/quantum_v1alpha1/services/quantum_engine_service/transports/grpc_asyncio.py

+3-2
Original file line numberDiff line numberDiff line change
@@ -214,8 +214,9 @@ def __init__(
214214
ssl_credentials=self._ssl_channel_credentials,
215215
quota_project_id=quota_project_id,
216216
options=[
217-
("grpc.max_send_message_length", -1),
218-
("grpc.max_receive_message_length", -1),
217+
('grpc.max_send_message_length', 20 * 1024 * 1024), # 20MiB
218+
('grpc.max_receive_message_length', -1), # unlimited
219+
('grpc.max_metadata_length', 10 * 1024 * 1024), # 10MiB
219220
],
220221
)
221222

cirq-google/cirq_google/engine/stream_manager.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -123,7 +123,7 @@ async def _make_request_queue(self) -> asyncio.Queue[Optional[quantum.QuantumRun
123123
124124
If `None` is put into the queue, the request iterator will stop.
125125
"""
126-
return asyncio.Queue()
126+
return asyncio.Queue(maxsize=100)
127127

128128
def submit(
129129
self, project_name: str, program: quantum.QuantumProgram, job: quantum.QuantumJob

0 commit comments

Comments
 (0)