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.gpufor GPU selection).- Parameters:
major (int)
minor (int)
release (int)
- Return type: