Unverified Commit a2d2acb4 authored by Chen Zhang's avatar Chen Zhang Committed by GitHub
Browse files

[Bugfix][Kernel] Give unique name to BlockSparseFlashAttention (#12040)


Signed-off-by: default avatarChen Zhang <zhangch99@outlook.com>
parent 2e0e0176
......@@ -89,8 +89,7 @@ class BlocksparseFlashAttentionBackend(AttentionBackend):
@staticmethod
def get_name() -> str:
# For attention layer compatibility
return "FLASH_ATTN"
return "BLOCK_SPARSE_FLASH_ATTN"
@staticmethod
def get_impl_cls() -> Type["BlocksparseFlashAttentionImpl"]:
......
......@@ -33,6 +33,7 @@ class _Backend(enum.Enum):
HPU_ATTN = enum.auto()
PALLAS = enum.auto()
IPEX = enum.auto()
BLOCK_SPARSE_FLASH_ATTN = enum.auto()
NO_ATTENTION = enum.auto()
......
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