Commit 660af62e authored by zhuwenwen's avatar zhuwenwen
Browse files

update rocm.py

parent c7096363
...@@ -201,9 +201,9 @@ class RocmPlatform(Platform): ...@@ -201,9 +201,9 @@ class RocmPlatform(Platform):
# f"does not support block size {block_size}." # f"does not support block size {block_size}."
# "(currently only supports block size 1)") # "(currently only supports block size 1)")
# else: # else:
raise ValueError( # raise ValueError(
f" The selected backend, {selected_backend.name}," # f" The selected backend, {selected_backend.name},"
f"is not MLA type while requested for MLA backend.") # f"is not MLA type while requested for MLA backend.")
selected_backend = (_Backend.ROCM_FLASH if selected_backend selected_backend = (_Backend.ROCM_FLASH if selected_backend
== _Backend.FLASH_ATTN else selected_backend) == _Backend.FLASH_ATTN else selected_backend)
......
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