def register_acc_op(acc_op: Callable):
    """
    For a new acc op, add this as decorator to register it.
    """
    _acc_ops.add(acc_op)
    return acc_op
For a new acc op, add this as decorator to register it.
fx2ait/fx2ait/acc_tracer/acc_normalizer.py