(op, use_fake_rule_for_debug=False)
| 22 | |
| 23 | |
| 24 | def get_rule(op, use_fake_rule_for_debug=False): |
| 25 | op_key = op if isinstance(op, str) else type(op) |
| 26 | if use_fake_rule_for_debug: |
| 27 | if op_key in lower_rule: |
| 28 | return lower_rule[op_key] |
| 29 | else: |
| 30 | warnings.warn(f"op: {op_key} not register, use fake op rule") |
| 31 | return lower_rule["fake_op_rule_for_debug"] |
| 32 | else: |
| 33 | return lower_rule[op_key] |
| 34 | |
| 35 | |
| 36 | def _log_mge_opr_attrs(mopr): |