Whether fp4 support is provided in the specified compute capability or not Parameters ---------- compute_version : str GPU capability
(compute_version)
| 1147 | |
| 1148 | @tvm_ffi.register_global_func("tvm.support.nvcc.supports_fp4") |
| 1149 | def have_fp4(compute_version): |
| 1150 | """Whether fp4 support is provided in the specified compute capability or not |
| 1151 | |
| 1152 | Parameters |
| 1153 | ---------- |
| 1154 | compute_version : str |
| 1155 | GPU capability |
| 1156 | """ |
| 1157 | major, minor = parse_compute_version(compute_version) |
| 1158 | # fp4 is suppored in Blackwell (10.0) or later architectures. |
| 1159 | if major == 10 and minor == 0: |
| 1160 | return True |
| 1161 | return False |
nothing calls this directly
no test coverage detected
searching dependent graphs…