Enable FP8 compute on AMD

This commit is contained in:
Your Name 2025-09-18 10:06:39 +03:00
parent dd611a7700
commit 9123ec20c3

View File

@ -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)