diff --git a/comfy/flash_attn_triton_amd/utils.py b/comfy/flash_attn_triton_amd/utils.py index 3e679906e..f1d7cfead 100644 --- a/comfy/flash_attn_triton_amd/utils.py +++ b/comfy/flash_attn_triton_amd/utils.py @@ -276,5 +276,5 @@ def is_cdna(): def is_rdna(): - return is_hip() and triton.runtime.driver.active.get_current_target().arch in ("gfx1030", "gfx1031", "gfx1100", "gfx1101", - "gfx1102", "gfx1200", "gfx1201") + return is_hip() and triton.runtime.driver.active.get_current_target().arch in ("gfx1030", "gfx1031", "gfx1035", "gfx1100", "gfx1101", + "gfx1102", "gfx1103", "gfx1150", "gfx1200", "gfx1201")