[ROCm] Split AITER unified attention into its own backend (#25507)

Signed-off-by: Gregory Shtrasberg <Gregory.Shtrasberg@amd.com>
This commit is contained in:
Gregory Shtrasberg
2025-10-06 18:49:23 -04:00
committed by GitHub
parent 2161efe978
commit f231e5bc21
8 changed files with 325 additions and 301 deletions

View File

@@ -254,3 +254,4 @@ def global_force_attn_backend_context_manager(
finally:
# Revert the original global backend override, if any
global_force_attn_backend(original_value)
_cached_get_attn_backend.cache_clear()