We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 479d4c9 commit d036445Copy full SHA for d036445
tests/unittest/_torch/executor/test_pytorch_model_engine.py
@@ -158,12 +158,12 @@ def test_pad_generation_requests(self) -> None:
158
resource_manager = ResourceManager(
159
{ResourceManagerType.KV_CACHE_MANAGER: kv_cache_manager})
160
161
- seqlens_and_batch_sizes = [
+ batch_sizes_and_seqlens = [
162
(5, 1),
163
(13, 1),
164
(5, 25),
165
]
166
- for (batch_size, max_seq_len) in seqlens_and_batch_sizes:
+ for (batch_size, max_seq_len) in batch_sizes_and_seqlens:
167
requests = [
168
_create_request(max_seq_len, i) for i in range(batch_size)
169
0 commit comments