Skip to content

Commit adb7259

Browse files
committed
fix ci
1 parent 43f9380 commit adb7259

File tree

1 file changed

+2
-5
lines changed

1 file changed

+2
-5
lines changed

vllm_ascend/attention/attention_v1.py

Lines changed: 2 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -218,9 +218,6 @@ def build_dummy_metadata(self, num_actual_tokens, num_reqs,
218218
num_scheduled_tokens, attn_state):
219219
if attn_state == AscendAttentionState.DecodeOnly:
220220
# NOTE: We only need to pay attention to seq_lens_list and block_table here
221-
common_attn_metadata = AscendCommonAttentionMetadata(
222-
seq_lens=torch.empty_like(self.runner.seq_lens_cpu).fill_(2))
223-
224221
block_table = self.runner.input_batch.block_table[0].block_table
225222
block_table[:num_reqs, 0] = torch.arange(1,
226223
num_reqs + 1,
@@ -231,8 +228,8 @@ def build_dummy_metadata(self, num_actual_tokens, num_reqs,
231228
block_table[:num_reqs, :self.runner.max_num_blocks_per_req] = (
232229
block_table[:num_reqs])
233230

234-
query_start_loc = common_attn_metadata.query_start_loc
235-
seq_lens = common_attn_metadata.seq_lens
231+
query_start_loc = None
232+
seq_lens = torch.empty_like(self.runner.seq_lens_cpu).fill_(2)
236233
query_lens = self.runner.query_lens
237234
seq_lens_list = None
238235

0 commit comments

Comments
 (0)