(self, op: Evaluate)
| 178 | self.body = body |
| 179 | |
| 180 | def visit_evaluate_(self, op: Evaluate): |
| 181 | value = op.value |
| 182 | if isinstance(value, Call) and value.op.same_as(Op.get("tirx.tvm_kernel_replace_point")): |
| 183 | return self.body |
| 184 | return super().visit_evaluate_(op) |
| 185 | |
| 186 | |
| 187 | def seek_kernel_replace_point(stmt: Stmt, body: Stmt) -> Stmt: |