Skip to content

Commit 13d2580

Browse files
authored
Enable PyTorch SDPA attention and FP8 ops for gfx1171 (#16039)
1 parent 2035799 commit 13d2580

1 file changed

Lines changed: 2 additions & 2 deletions

File tree

comfy/model_management.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -519,13 +519,13 @@ def aotriton_supported():
519519
if args.use_split_cross_attention == False and args.use_quad_cross_attention == False:
520520
if aotriton_supported(): # AMD efficient attention implementation depends on aotriton.
521521
if torch_version_numeric >= (2, 7): # works on 2.6 but doesn't actually seem to improve much
522-
if any((a in arch) for a in ["gfx90a", "gfx942", "gfx950", "gfx1100", "gfx1101", "gfx1150", "gfx1151", "gfx1170"]): # TODO: more arches, TODO: gfx950
522+
if any((a in arch) for a in ["gfx90a", "gfx942", "gfx950", "gfx1100", "gfx1101", "gfx1150", "gfx1151", "gfx1170", "gfx1171"]): # TODO: more arches, TODO: gfx950
523523
ENABLE_PYTORCH_ATTENTION = True
524524
if rocm_version >= (7, 0):
525525
if any((a in arch) for a in ["gfx1200", "gfx1201"]):
526526
ENABLE_PYTORCH_ATTENTION = True
527527
if torch_version_numeric >= (2, 7) and rocm_version >= (6, 4):
528-
if any((a in arch) for a in ["gfx1200", "gfx1201", "gfx950", "gfx1170"]): # TODO: more arches, "gfx942" gives error on pytorch nightly 2.10 1013 rocm7.0
528+
if any((a in arch) for a in ["gfx1200", "gfx1201", "gfx950", "gfx1170", "gfx1171"]): # TODO: more arches, "gfx942" gives error on pytorch nightly 2.10 1013 rocm7.0
529529
SUPPORT_FP8_OPS = True
530530

531531
except:

0 commit comments

Comments
 (0)