wangbo/ComfyUI
1
0
Fork 0
mirror of https://github.com/comfyanonymous/ComfyUI.git synced 2026-01-10 06:10:50 +08:00
Code Issues Actions 10 Packages Projects Releases Wiki Activity

This will allow much better support for gfx1032 and other things not specifically named

Browse Source
This commit is contained in:
Christopher Anderson 2025-07-29 04:21:45 +10:00
parent d42553749d
commit b5ede18481
1 changed files with 2 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
comfy/flash_attn_triton_amd/utils.py
View File

@ -269,12 +269,8 @@ def get_input_shapes():
def is_hip():
return triton.runtime.driver.active.get_current_target().backend == "hip"
def is_cdna():
return is_hip() and triton.runtime.driver.active.get_current_target().arch in ('gfx940', 'gfx941', 'gfx942',
'gfx90a', 'gfx908')
return is_hip() and triton.runtime.driver.active.get_current_target().arch.startswith('gfx9')
def is_rdna():
return is_hip() and triton.runtime.driver.active.get_current_target().arch in ("gfx1030", "gfx1031", "gfx1035", "gfx1100", "gfx1101",
"gfx1102", "gfx1103", "gfx1150", "gfx1200", "gfx1201")
return is_hip() and triton.runtime.driver.active.get_current_target().arch.startswith("gfx1")