Convert a raw version code string (e.g. '118', '713') to a dotted version (e.g. '11.8', '7.13').
(version_str: str)
| 157 | |
| 158 | |
| 159 | def parse_cuda_version(version_str: str) -> str: |
| 160 | """Convert a raw version code string (e.g. '118', '713') to a dotted version (e.g. '11.8', '7.13').""" |
| 161 | if version_str.isdigit(): |
| 162 | is_hip = bool(torch.version.hip) |
| 163 | try: |
| 164 | major, minor = _split_cuda_version(version_str, is_hip) |
| 165 | return f"{major}.{minor}" |
| 166 | except (ValueError, IndexError): |
| 167 | pass |
| 168 | return version_str |
| 169 | |
| 170 | |
| 171 | class ErrorHandlerMockBNBNativeLibrary(BNBNativeLibrary): |
no test coverage detected