tvm_tirx.support.nvcc.parse_compute_version

tvm_tirx.support.nvcc.parse_compute_version(compute_version)

Parse compute capability string to divide major and minor version

Parameters:

compute_version (str) – compute capability of a GPU (e.g. “6.0”)

Returns:

• major (int) – major version number

• minor (int) – minor version number