Skip to content

Commit 2a6d53c

Browse files
committed
ci: Ensemble requests stuck indefinitely when the step max_queue_size exceeded (#8631)
1 parent 868eb15 commit 2a6d53c

4 files changed

Lines changed: 348 additions & 51 deletions

File tree

qa/L0_simple_ensemble/backpressure_test_models/decoupled_producer/1/model.py

Lines changed: 6 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -1,4 +1,4 @@
1-
# Copyright 2025, NVIDIA CORPORATION & AFFILIATES. All rights reserved.
1+
# Copyright 2025-2026, NVIDIA CORPORATION & AFFILIATES. All rights reserved.
22
#
33
# Redistribution and use in source and binary forms, with or without
44
# modification, are permitted provided that the following conditions
@@ -25,6 +25,8 @@
2525
# OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
2626

2727

28+
import time
29+
2830
import numpy as np
2931
import triton_python_backend_utils as pb_utils
3032

@@ -38,13 +40,14 @@ def execute(self, requests):
3840
for request in requests:
3941
# Get input - number of responses to produce
4042
in_tensor = pb_utils.get_input_tensor_by_name(request, "IN")
41-
count = in_tensor.as_numpy()[0]
43+
count = in_tensor.as_numpy().item()
4244

4345
response_sender = request.get_response_sender()
46+
out_tensor = pb_utils.Tensor("OUT", np.array([[0.5]], dtype=np.float32))
4447

4548
# Produce 'count' responses, each with 0.5 as the output value
4649
for i in range(count):
47-
out_tensor = pb_utils.Tensor("OUT", np.array([0.5], dtype=np.float32))
50+
time.sleep(0.1) # Simulate some processing delay
4851
response = pb_utils.InferenceResponse(output_tensors=[out_tensor])
4952
response_sender.send(response)
5053

qa/L0_simple_ensemble/backpressure_test_models/decoupled_producer/config.pbtxt

Lines changed: 2 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -1,4 +1,4 @@
1-
# Copyright 2025, NVIDIA CORPORATION & AFFILIATES. All rights reserved.
1+
# Copyright 2025-2026, NVIDIA CORPORATION & AFFILIATES. All rights reserved.
22
#
33
# Redistribution and use in source and binary forms, with or without
44
# modification, are permitted provided that the following conditions
@@ -27,7 +27,7 @@
2727

2828
name: "decoupled_producer"
2929
backend: "python"
30-
max_batch_size: 0
30+
max_batch_size: 1
3131

3232
input [
3333
{
@@ -55,4 +55,3 @@ instance_group [
5555
model_transaction_policy {
5656
decoupled: true
5757
}
58-

0 commit comments

Comments
 (0)