format

shen-shanshan · shen-shanshan · commit d0c1e2a28bef · 2025-07-31T06:57:54.000Z
Signed-off-by: shen-shanshan &lt;467638484@qq.com&gt;
diff --git a/vllm_ascend/attention/attention_v1.py b/vllm_ascend/attention/attention_v1.py
@@ -347,6 +347,7 @@ def _forward_prefill_no_cache(
                                        num_heads=self.num_heads,
                                        num_kv_heads=self.num_kv_heads,
                                        out=output)
+        assert output is not None
         return output[:num_tokens, :, :]
 
     def _forward_prefill_cache_hit(
