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

update rocm.py

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