cleanup boolean logic

Signed-off-by: Sage Moore <[email protected]>

Author: SageMoore

vllm/attention/backends/mla/common.py

@@ -1323,8 +1323,7 @@ def _compute_prefill_context(
                                                [0, q.shape[-1] - v.shape[-1]],
                                                value=0)
 
-            if is_hip and envs.VLLM_USE_TRITON_FLASH_ATTN and \
-                has_context is False:
+            if is_hip and envs.VLLM_USE_TRITON_FLASH_ATTN and not has_context:
                 attn_output, attn_softmax_lse = self.triton_fa_func(
                     q,
                     k,
@@ -1413,7 +1412,7 @@ def _forward_prefill(
         v_padded = torch.nn.functional.pad(v, [0, q.shape[-1] - v.shape[-1]],
                                            value=0)
 
-        if is_hip and envs.VLLM_USE_TRITON_FLASH_ATTN and has_context is False:
+        if is_hip and envs.VLLM_USE_TRITON_FLASH_ATTN and not has_context:
             output = self.triton_fa_func(
                 q,
                 k,