We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 9d79b8d commit 080d391Copy full SHA for 080d391
thunder/executors/nvfuserex_impl.py
@@ -161,8 +161,8 @@ def is_supported_devicetype(devicetype: DeviceType) -> bool:
161
162
163
def device_supports_fp8() -> bool:
164
- cuda_major, _ = torch.cuda.get_device_capability()
165
- return cuda_major > 8
+ cuda_major, cuda_minor = torch.cuda.get_device_capability()
+ return (cuda_major, cuda_minor) >= (8, 9)
166
167
168
def is_supported_dtype(dtype: type | dtypes.dtype, *, allow_low_precision_floats: bool = True) -> bool:
0 commit comments