tvm_tirx.codegen

tvm_tirx.codegen#

Compile IR modules into executable runtime modules and query target features.

tvm_tirx.codegen.build_module(mod, target)#

Build IRModule into Module.

Parameters:
  • mod (tvm_tirx.IRModule) – The ir module.

  • target (str) – The target module type.

Returns:

module – The corresponding module.

Return type:

tvm_ffi.Module

tvm_tirx.codegen.target_has_features(cpu_features, target=None)#

Check CPU features for the target’s -mtriple and -mcpu and -mattr.

Parameters:
  • target (Target) – The TVM target.

  • cpu_features (str or Array) – CPU Feature(s) to check.

Returns:

has_features – True if target has the feature(s).

Return type:

bool