Either bf16 support is provided in the compute capability or not Parameters ---------- compute_version : str compute capability of a GPU (e.g. "8.0")
(compute_version)
| 1113 | |
| 1114 | @tvm_ffi.register_global_func("tvm.support.nvcc.supports_bf16") |
| 1115 | def have_bf16(compute_version): |
| 1116 | """Either bf16 support is provided in the compute capability or not |
| 1117 | |
| 1118 | Parameters |
| 1119 | ---------- |
| 1120 | compute_version : str |
| 1121 | compute capability of a GPU (e.g. "8.0") |
| 1122 | """ |
| 1123 | major, _ = parse_compute_version(compute_version) |
| 1124 | if major >= 8: |
| 1125 | return True |
| 1126 | |
| 1127 | return False |
| 1128 | |
| 1129 | |
| 1130 | @tvm_ffi.register_global_func("tvm.support.nvcc.supports_fp8") |
searching dependent graphs…