tvm_tirx.tirx.register_intrin_lowering

tvm_tirx.tirx.register_intrin_lowering#

tvm_tirx.tirx.register_intrin_lowering(op_name, target, *, f=None, override=False)#

Register Op lowering function

Parameters:
  • op_name (str) – The op name

  • target (str) – The target string for given intrinsic lowering function

  • f (function, optional) – The function to be registered.

  • override (bool, optional) – Replace an existing lowering if True; duplicate registration otherwise raises ValueError.

Returns:

result (function) – The registered lowering, or a decorator if f is not supplied.