From 89c4f78c59dd6c2777329c6f3462b2e45c724337 Mon Sep 17 00:00:00 2001
From: Kunshang Ji <kunshang.ji@intel.com>
Date: Wed, 27 Nov 2024 14:49:40 +0800
Subject: [PATCH] [Hardware][Gaudi]add get_name method for HPUAttentionBackend
 (#10667)

Signed-off-by: Kunshang Ji <kunshang.ji@intel.com>
Signed-off-by: Andrew Feldman <afeldman@neuralmagic.com>
---
 vllm/attention/backends/hpu_attn.py | 4 ++++
 1 file changed, 4 insertions(+)

diff --git a/vllm/attention/backends/hpu_attn.py b/vllm/attention/backends/hpu_attn.py
index 4a3ddd5db94e5..5359941d41fde 100644
--- 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