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.

参数:

• 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.

返回:

hsaco -- The bytearray of the hsaco

返回类型:

bytearray

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

use hipcc to generate fatbin code for better optimization