fix hpu_attn get_name

Signed-off-by: Kunshang Ji <[email protected]>
jikunshang · Nov 27, 2024 · ab9658d · ab9658d
1 parent 9200e9a
commit ab9658d
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/vllm/attention/backends/hpu_attn.py b/vllm/attention/backends/hpu_attn.py
@@ -22,6 +22,10 @@
 
 class HPUAttentionBackend(AttentionBackend):
 
+    @staticmethod
+    def get_name() -> str:
+        return "HPU_ATTN"
+
     @staticmethod
     def get_impl_cls() -> Type["HPUAttentionImpl"]:
         return HPUAttentionImpl