add triton_key

vllm/attention/backends/rocm_flash_attn.py

@@ -5,6 +5,8 @@ from dataclasses import dataclass
 from typing import TYPE_CHECKING, Any, Dict, List, Optional, Tuple, Type
 
 import torch
+import triton
+from triton.compiler.compiler import triton_key
 
 import vllm.envs as envs
 from vllm import _custom_ops as ops
@@ -778,6 +780,7 @@ class ROCmFlashAttentionImpl(AttentionImpl):
             else:
                 # prefix-enabled attention
                 # not applicable for encoder-only models
+                version_key = triton_key()
                 if self.attn_type != AttentionType.ENCODER_ONLY:
                     output[:
                            num_prefill_tokens] = PagedAttention.forward_prefix(