tvm_tirx.testing.env.has_cuda_compute

tvm_tirx.testing.env.has_cuda_compute(major, minor=0, exact=False)

True if the CUDA compute capability satisfies (major, minor).

When exact is False (default) the check is compute >= (major, minor); when True it requires an exact match. Returns False when no CUDA device is present, so it implies has_cuda().

Parameters:

• major (int)

• minor (int)

• exact (bool)

Return type:

bool