(expr, op_name: str)
| 81 | ) |
| 82 | |
| 83 | def _check_call(expr, op_name: str): |
| 84 | assert isinstance(expr, rx.Call) |
| 85 | if not op_name.startswith("relax."): |
| 86 | op_name = "relax." + op_name |
| 87 | op = tvm.ir.Op.get(op_name) |
| 88 | assert expr.op == op |
| 89 | |
| 90 | # Comparison operators |
| 91 | _check_call(x > y, "greater") |
no test coverage detected
searching dependent graphs…