[Hardware][Gaudi]add get_name method for HPUAttentionBackend (#10667)

Signed-off-by: Kunshang Ji <kunshang.ji@intel.com>
This commit is contained in:
Kunshang Ji authored and GitHub committed 2024-11-26 22:49:40 -08:00
1 parent cfb3bf25fb
commit e85250b1d1
1 file changed
+4
+4
View File
@@ -22,6 +22,10 @@ logger = init_logger(__name__)
class HPUAttentionBackend(AttentionBackend):
@staticmethod
def get_name() -> str:
return "HPU_ATTN"
@staticmethod
def get_impl_cls() -> Type["HPUAttentionImpl"]:
return HPUAttentionImpl