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 40f7972 commit 28af668Copy full SHA for 28af668
thunder/executors/nvfuserex_impl.py
@@ -164,8 +164,8 @@ def is_supported_devicetype(devicetype: DeviceType) -> bool:
164
165
166
def device_supports_fp8() -> bool:
167
- cuda_major, _ = torch.cuda.get_device_capability()
168
- return cuda_major > 8
+ cuda_major, cuda_minor = torch.cuda.get_device_capability()
+ return (cuda_major, cuda_minor) >= (8, 9)
169
170
171
def is_supported_dtype(dtype: type | dtypes.dtype, *, allow_low_precision_floats: bool = True) -> bool:
0 commit comments