| 80 | |
| 81 | @dataclasses.dataclass |
| 82 | class LoweringRuleContext: |
| 83 | module_context: ModuleContext |
| 84 | op: OpInfo |
| 85 | vars_in: Sequence[VarInfo] |
| 86 | vars_out: Sequence[VarInfo] |
| 87 | param: Dict = None |
| 88 | |
| 89 | def replace(self, **kw): |
| 90 | return dataclasses.replace(self, **kw) |
| 91 | |
| 92 | |
| 93 | def _unwrap_singleton_ir_values(x): |