mirror of
https://git.datalinker.icu/comfyanonymous/ComfyUI
synced 2026-08-14 09:10:05 +08:00
Enable FP8 compute on AMD
This commit is contained in:
parent
dd611a7700
commit
9123ec20c3
@ -1355,7 +1355,7 @@ def supports_fp8_compute(device=None):
|
||||
if SUPPORT_FP8_OPS:
|
||||
return True
|
||||
|
||||
if not is_nvidia():
|
||||
if not (is_nvidia() or is_amd()):
|
||||
return False
|
||||
|
||||
props = torch.cuda.get_device_properties(device)
|
||||
|
||||
Loading…
x
Reference in New Issue
Block a user