Skip to content

Commit 55ad364

Browse files
rename
Signed-off-by: Matthew Bonanni <mbonanni@redhat.com>
1 parent 3ba013e commit 55ad364

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

vllm/attention/backends/registry.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -29,7 +29,7 @@ class _Backend(enum.Enum):
2929
TREE_ATTN = enum.auto()
3030

3131

32-
BACKEND_MAPPING = {}
32+
BACKEND_MAP = {}
3333

3434

3535
def register_attn_backend(backend: _Backend, class_path: str | None = None):
@@ -45,7 +45,7 @@ def register_attn_backend(backend: _Backend, class_path: str | None = None):
4545

4646
def decorator(cls):
4747
path = class_path or f"{cls.__module__}.{cls.__qualname__}"
48-
BACKEND_MAPPING[backend] = path
48+
BACKEND_MAP[backend] = path
4949
return cls
5050

5151
return decorator
@@ -60,7 +60,7 @@ def backend_to_class_str(backend: _Backend) -> str:
6060
Returns:
6161
The backend class string
6262
"""
63-
return BACKEND_MAPPING[backend]
63+
return BACKEND_MAP[backend]
6464

6565

6666
def backend_to_class(backend: _Backend) -> Type:

0 commit comments

Comments
 (0)