Skip to content

GitLab

Unverified Commit e85250b1 authored by Kunshang Ji's avatar Kunshang Ji Committed by GitHub
Browse files

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


Signed-off-by: default avatarKunshang Ji <kunshang.ji@intel.com>
parent cfb3bf25
Hide whitespace changes
Inline Side-by-side
Showing with 4 additions and 0 deletions
vllm/attention/backends/hpu_attn.py
View file @ e85250b1
...@@ -22,6 +22,10 @@ logger = init_logger(__name__) ...@@ -22,6 +22,10 @@ logger = init_logger(__name__)
class HPUAttentionBackend(AttentionBackend): class HPUAttentionBackend(AttentionBackend):
@staticmethod
def get_name() -> str:
return "HPU_ATTN"
@staticmethod @staticmethod
def get_impl_cls() -> Type["HPUAttentionImpl"]: def get_impl_cls() -> Type["HPUAttentionImpl"]:
return HPUAttentionImpl return HPUAttentionImpl
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment