tvm_tirx.testing.env.has_nvcc_version

tvm_tirx.testing.env.has_nvcc_version#

tvm_tirx.testing.env.has_nvcc_version(major, minor=0, release=0)#

True if a CUDA device is present and nvcc is at least (major, minor, release).

Returns False when no CUDA device is present, so it implies has_cuda(). Gate a test with @pytest.mark.skipif(not tvm_tirx.testing.env.has_nvcc_version(11, 4), reason="need nvcc >= 11.4") (add @pytest.mark.gpu for GPU selection).

Parameters:
  • major (int)

  • minor (int)

  • release (int)

Return type:

bool