tvm_tirx.support.rocm

Contents

tvm_tirx.support.rocm#

Utility for ROCm backend

Functions#

find_lld([required])

Find ld.lld in system.

rocm_link(in_file, out_file[, lld])

Link relocatable ELF object to shared ELF object using lld

callback_rocm_link

Callable wrapper around a TVM FFI function.

callback_rocm_bitcode_path

Callable wrapper around a TVM FFI function.

parse_compute_version(compute_version)

Parse compute capability string to divide major and minor version

have_matrixcore([compute_version])

Either MatrixCore support is provided in the compute capability or not

get_rocm_arch

Callable wrapper around a TVM FFI function.

find_rocm_path()

Utility function to find ROCm path
Contents