Either int8 support is provided in the compute capability or not Parameters ---------- compute_version : str compute capability of a GPU (e.g. "6.1")
(compute_version)
| 1053 | |
| 1054 | |
| 1055 | def have_int8(compute_version): |
| 1056 | """Either int8 support is provided in the compute capability or not |
| 1057 | |
| 1058 | Parameters |
| 1059 | ---------- |
| 1060 | compute_version : str |
| 1061 | compute capability of a GPU (e.g. "6.1") |
| 1062 | """ |
| 1063 | major, _ = parse_compute_version(compute_version) |
| 1064 | if major >= 6: |
| 1065 | return True |
| 1066 | |
| 1067 | return False |
| 1068 | |
| 1069 | |
| 1070 | def have_tensorcore(compute_version=None, target=None): |
searching dependent graphs…