Whether fp8 support is provided in the specified compute capability or not Parameters ---------- compute_version : str GPU capability
(compute_version)
| 1129 | |
| 1130 | @tvm_ffi.register_global_func("tvm.support.nvcc.supports_fp8") |
| 1131 | def have_fp8(compute_version): |
| 1132 | """Whether fp8 support is provided in the specified compute capability or not |
| 1133 | |
| 1134 | Parameters |
| 1135 | ---------- |
| 1136 | compute_version : str |
| 1137 | GPU capability |
| 1138 | """ |
| 1139 | major, minor = parse_compute_version(compute_version) |
| 1140 | # fp8 is suppored in Ada Lovelace (8.9) or later architectures. |
| 1141 | if major == 8 and minor == 9: |
| 1142 | return True |
| 1143 | if major >= 9: |
| 1144 | return True |
| 1145 | return False |
| 1146 | |
| 1147 | |
| 1148 | @tvm_ffi.register_global_func("tvm.support.nvcc.supports_fp4") |
nothing calls this directly
no test coverage detected
searching dependent graphs…