tvm_tirx.support.nvcc.have_tensorcore

tvm_tirx.support.nvcc.have_tensorcore(compute_version=None, target=None)

Either TensorCore support is provided in the compute capability or not

Parameters:

• compute_version (str, optional) – compute capability of a GPU (e.g. “7.0”).

• target (tvm_tirx.target.Target, optional) – The compilation target, will be used to determine arch if compute_version isn’t specified.