[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
2024-11-26 22:49:40 -08:00
committed by GitHub
parent cfb3bf25fb
commit e85250b1d1
+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