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.