Add for _BINARY_OP

Fidget-Spinner · Fidget-Spinner · commit 634ed26c8d6a · 2025-06-20T19:11:11.000+08:00
diff --git a/Python/optimizer_bytecodes.c b/Python/optimizer_bytecodes.c
@@ -181,6 +181,7 @@ dummy_func(void) {
     }
 
     op(_BINARY_OP, (lhs, rhs -- res)) {
+        REPLACE_OPCODE_IF_EVALUATES_PURE(lhs, rhs);
         bool lhs_int = sym_matches_type(lhs, &PyLong_Type);
         bool rhs_int = sym_matches_type(rhs, &PyLong_Type);
         bool lhs_float = sym_matches_type(lhs, &PyFloat_Type);
diff --git a/Python/optimizer_cases.c.h b/Python/optimizer_cases.c.h
diff --git a/Tools/cases_generator/optimizer_generator.py b/Tools/cases_generator/optimizer_generator.py
@@ -170,11 +170,6 @@ def replace_opcode_if_evaluates_pure(
     ) -> bool:
         skip_to(tkn_iter, "SEMI")
 
-        if self.original_uop.properties.escapes:
-            raise analysis_error(
-                f"REPLACE_OPCODE_IF_EVALUATES_PURE cannot be used with an escaping uop {self.original_uop.properties.escaping_calls}",
-                self.original_uop.body.open
-            )
         emitter = OptimizerConstantEmitter(self.out, {}, self.original_uop, copy.deepcopy(self.stack))
         emitter.emit("if (\n")
         assert isinstance(uop, Uop)

Original file line number	Diff line number	Diff line change
`@@ -181,6 +181,7 @@ dummy_func(void) {`
`181`	`181`	`}`
`182`	`182`
`183`	`183`	`op(_BINARY_OP, (lhs, rhs -- res)) {`
	`184`	`+ REPLACE_OPCODE_IF_EVALUATES_PURE(lhs, rhs);`
`184`	`185`	`bool lhs_int = sym_matches_type(lhs, &PyLong_Type);`
`185`	`186`	`bool rhs_int = sym_matches_type(rhs, &PyLong_Type);`
`186`	`187`	`bool lhs_float = sym_matches_type(lhs, &PyFloat_Type);`