mirror of
https://github.com/comfyanonymous/ComfyUI.git
synced 2026-02-17 00:43:48 +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…
Reference in New Issue
Block a user