get_hip_arch_name -> get_hip_agent, use amdhip64_7.dll served within rocm package

This commit is contained in:
Seunghoon Lee
2026-01-03 21:00:36 +09:00
parent 0158e61814
commit 49965dfda8
7 changed files with 138 additions and 135 deletions
+2 -1
View File
@@ -52,7 +52,8 @@ def get_default_modes(cmd_opts, mem_stat):
default_sdp_override_choices.append('Triton Flash attention')
elif devices.backend == "rocm":
default_sdp_override_choices.append('Triton Flash attention')
if int(devices.get_hip_arch_name()) < 1100:
agent = devices.get_hip_agent()
if agent.gfx_version < 0x1100:
default_sdp_override_options = ['Dynamic attention'] # only RDNA2 and older GPUs needs this
elif devices.backend in {"directml", "cpu", "mps"}:
default_sdp_override_options = ['Dynamic attention']