tilelang.contrib.hipcc

Utility to invoke hipcc compiler in the system

函式

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