tilelang.contrib.rocm module#

Utility for ROCm backend

tilelang.contrib.rocm.find_lld(required=True)#

Find ld.lld in system.

Parameters:

required (bool) – Whether it is required, runtime error will be raised if the compiler is required.

Returns:

valid_list – List of possible paths.

Return type:

list of str

Note

This function will first search ld.lld that matches the major llvm version that built with tvm

tilelang.contrib.rocm.find_rocm_path()#

Utility function to find ROCm path

Returns:

path – Path to ROCm root.

Return type:

str

tilelang.contrib.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 – True if MatrixCore support is provided, False otherwise

Return type:

bool

tilelang.contrib.rocm.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

Link relocatable ELF object to shared ELF object using lld

Parameters:

  • in_file (str) – Input file name (relocatable ELF object file)

  • out_file (str) – Output file name (shared ELF object file)

  • lld (str, optional) – The lld linker, if not specified, we will try to guess the matched clang version.