Skip to content

Commit 56f5fd9

Browse files
0xDELUXAadlerfaulkner
authored andcommitted
Enable fp8 ops by default on gfx1200 (Comfy-Org#9926)
1 parent 1443f7e commit 56f5fd9

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

comfy/model_management.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -348,7 +348,7 @@ def amd_min_version(device=None, min_rdna_version=0):
348348
# if any((a in arch) for a in ["gfx1201"]):
349349
# ENABLE_PYTORCH_ATTENTION = True
350350
if torch_version_numeric >= (2, 7) and rocm_version >= (6, 4):
351-
if any((a in arch) for a in ["gfx1201", "gfx942", "gfx950"]): # TODO: more arches
351+
if any((a in arch) for a in ["gfx1200", "gfx1201", "gfx942", "gfx950"]): # TODO: more arches
352352
SUPPORT_FP8_OPS = True
353353

354354
except:

0 commit comments

Comments
 (0)