gfx942 doesn't support fp8 operations. (#10348)

This commit is contained in:
comfyanonymous 2025-10-14 21:21:11 -07:00 committed by GitHub
parent ddfce1af4f
commit 1c10b33f9b
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

View File

@ -349,7 +349,7 @@ try:
if any((a in arch) for a in ["gfx1201"]):
ENABLE_PYTORCH_ATTENTION = True
if torch_version_numeric >= (2, 7) and rocm_version >= (6, 4):
if any((a in arch) for a in ["gfx1200", "gfx1201", "gfx942", "gfx950"]): # TODO: more arches
if any((a in arch) for a in ["gfx1200", "gfx1201", "gfx950"]): # TODO: more arches, "gfx942" gives error on pytorch nightly 2.10 1013 rocm7.0
SUPPORT_FP8_OPS = True
except: