检查GPU是否支持fp8
()
| 135 | |
| 136 | |
| 137 | def is_fp8_supported_gpu(): |
| 138 | """检查GPU是否支持fp8""" |
| 139 | if not torch.cuda.is_available(): |
| 140 | return False |
| 141 | compute_capability = torch.cuda.get_device_capability(0) |
| 142 | major, minor = compute_capability |
| 143 | return (major == 8 and minor == 9) or (major >= 9) |
| 144 | |
| 145 | |
| 146 | def is_sm_greater_than_90(): |
no test coverage detected