tilelang.contrib.hipcc

Utility to invoke hipcc compiler in the system

Functions

compile_hip(code[, target_format, arch, options, ...])	Compile HIP code with hipcc.
tilelang_callback_hip_compile(code, target)	use hipcc to generate fatbin code for better optimization

Module Contents

tilelang.contrib.hipcc.compile_hip(code, target_format='hsaco', arch=None, options=None, path_target=None, verbose=False)

Compile HIP code with hipcc.

Parameters:

• code (str) – The HIP code.

• target_format (str) – The target format of hipcc compiler.

• arch (str) – The AMD GPU architecture.

• options (str or list of str) – The additional options.

• path_target (str, optional) – Output file.

Returns:

hsaco – The bytearray of the hsaco

Return type:

bytearray

tilelang.contrib.hipcc.tilelang_callback_hip_compile(code, target)

use hipcc to generate fatbin code for better optimization