Return the (major, minor, release) nvcc version, or (0, 0, 0).
()
| 331 | |
| 332 | @functools.cache |
| 333 | def _nvcc_version() -> tuple: |
| 334 | """Return the (major, minor, release) nvcc version, or (0, 0, 0).""" |
| 335 | try: |
| 336 | from tvm.support import nvcc # pylint: disable=import-outside-toplevel |
| 337 | |
| 338 | return nvcc.get_cuda_version() |
| 339 | except Exception: # pylint: disable=broad-except |
| 340 | return (0, 0, 0) |
| 341 | |
| 342 | |
| 343 | def has_nvcc_version(major: int, minor: int = 0, release: int = 0) -> bool: |
no outgoing calls
no test coverage detected
searching dependent graphs…