MCPcopy Create free account
hub / github.com/apache/tvm / _nvcc_version

Function _nvcc_version

python/tvm/testing/env.py:333–340  ·  view source on GitHub ↗

Return the (major, minor, release) nvcc version, or (0, 0, 0).

()

Source from the content-addressed store, hash-verified

331
332@functools.cache
333def _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
343def has_nvcc_version(major: int, minor: int = 0, release: int = 0) -> bool:

Callers 1

has_nvcc_versionFunction · 0.85

Calls

no outgoing calls

Tested by

no test coverage detected

Used in the wild real call sites across dependent graphs

searching dependent graphs…