tvm_tirx.support.rocm.have_matrixcore

tvm_tirx.support.rocm.have_matrixcore(compute_version=None)

Either MatrixCore support is provided in the compute capability or not

Parameters:

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

Returns:

have_matrixcore (bool) – True if MatrixCore support is provided, False otherwise