[Bugfix][ROCm] Strip block_size before attention backend validation (#36274)

Signed-off-by: jennyyyyzhen <yzhen@hmc.edu>
Co-authored-by: Lu Fang <30275821+houseroad@users.noreply.github.com>
This commit is contained in:
jennyyyyzhen
2026-03-11 13:37:31 -07:00
committed by GitHub
parent ff1e3d9c63
commit 428bc718bd

View File

@@ -438,6 +438,8 @@ class RocmPlatform(Platform):
device_capability = cls.get_device_capability()
assert device_capability is not None
attn_selector_config = attn_selector_config._replace(block_size=None)
# First try checking just the selected backend, if there is one.
if selected_backend is not None:
try: